bug fix with imports

This commit is contained in:
Steve Nyemba 2020-01-04 23:02:15 -06:00
parent 4165fabe57
commit ef39969082
4 changed files with 14 additions and 2 deletions

View File

@ -1 +1,2 @@
import data.params as params

View File

@ -10,7 +10,7 @@ This package is designed to generate synthetic data from a dataset from an origi
"""
import pandas as pd
import numpy as np
from data import gan
import data.gan as gan
from transport import factory
def train (**args) :
"""

10
data/maker/__main__.py Normal file
View File

@ -0,0 +1,10 @@
import pandas as pd
import data.maker
df = pd.read_csv('sample.csv')
column = 'gender'
id = 'id'
context = 'demo'
store = {"type":"mongo.MongoWriter","args":{"host":"localhost:27017","dbname":"GAN"}}
max_epochs = 11
data.maker.train(store=store,max_epochs=max_epochs,context=context,data=df,column=column,id=id,logs='foo')

View File

@ -5,9 +5,10 @@ import sys
def read(fname):
return open(os.path.join(os.path.dirname(__file__), fname)).read()
args = {"name":"data-maker","version":"1.0.0","author":"Vanderbilt University Medical Center","author_email":"steve.l.nyemba@vanderbilt.edu","license":"MIT",
"packages":["data"],"keywords":["healthcare","data","transport","protocol"]}
"packages":find_packages(),"keywords":["healthcare","data","transport","protocol"]}
args["install_requires"] = ['data-transport@git+https://dev.the-phi.com/git/steve/data-transport.git','numpy','pandas','pandas-gbq','pymongo']
args['url'] = 'https://hiplab.mc.vanderbilt.edu/aou/gan.git'
if sys.version_info[0] == 2 :
args['use_2to3'] = False
args['use_2to3_exclude_fixers'] = ['lib2to3.fixes.fix_import']