From ff076013b9a018323b62a686dccd39e7e52e66eb Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Tue, 7 Jan 2020 10:32:36 -0600 Subject: [PATCH] bug fix, with number of gpu parameter --- data/maker/__init__.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/data/maker/__init__.py b/data/maker/__init__.py index 469a65a..967dbc8 100644 --- a/data/maker/__init__.py +++ b/data/maker/__init__.py @@ -28,7 +28,7 @@ def train (**args) : logs = args['logs'] real = pd.get_dummies(df[column]).astype(np.float32).values labels = pd.get_dummies(df[column_id]).astype(np.float32).values - + num_gpu = 1 if 'num_gpu' not in args else args['num_gpu'] max_epochs = 10 if 'max_epochs' not in args else args['max_epochs'] context = args['context'] if 'store' in args : diff --git a/setup.py b/setup.py index ecef3bd..aa45602 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ 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", +args = {"name":"data-maker","version":"1.0.1","author":"Vanderbilt University Medical Center","author_email":"steve.l.nyemba@vanderbilt.edu","license":"MIT", "packages":find_packages(),"keywords":["healthcare","data","transport","protocol"]} args["install_requires"] = ['data-transport@git+https://dev.the-phi.com/git/steve/data-transport.git','tensorflow==1.14.0','numpy==1.16.3','pandas','pandas-gbq','pymongo'] args['url'] = 'https://hiplab.mc.vanderbilt.edu/aou/data-maker.git'