From 65a1fadfcaeefc67df9b84ab5053103a72f218c6 Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Sun, 12 Apr 2020 20:07:15 -0500 Subject: [PATCH] bug fix data type and pipeline --- data/gan.py | 3 ++- pipeline.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/data/gan.py b/data/gan.py index a46740a..5975255 100644 --- a/data/gan.py +++ b/data/gan.py @@ -592,7 +592,8 @@ class Predict(GNet): # The code below will insure we have some acceptable cardinal relationships between id and synthetic values # - df = pd.DataFrame(np.round(f)).astype(np.int32) + # df = pd.DataFrame(np.round(f)).astype(np.int32) + df = pd.DataFrame(np.round(f),dtype=np.int32) p = 0 not in df.sum(axis=1).values x = df.sum(axis=1).values diff --git a/pipeline.py b/pipeline.py index 22c637d..c243ec3 100644 --- a/pipeline.py +++ b/pipeline.py @@ -125,9 +125,9 @@ class Components : logger.write({"module":"train","action":"train","input":info}) data.maker.train(**_args) - if set(['drone','autopilot']) in set( list(args.keys())) : + if 'autopilot' in ( list(args.keys())) : print (['drone mode enabled ....']) - self.generate(**args) + self.generate(args) pass