diff --git a/data/__init__.py b/data/__init__.py index 88bbded..98124f1 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -1 +1,2 @@ import data.params as params + diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 075bfd3..469a65a 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -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) : """ diff --git a/data/maker/__main__.py b/data/maker/__main__.py new file mode 100644 index 0000000..e77bf0a --- /dev/null +++ b/data/maker/__main__.py @@ -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') \ No newline at end of file diff --git a/setup.py b/setup.py index 1bfc141..a7befdf 100644 --- a/setup.py +++ b/setup.py @@ -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']