bug fix data type and pipeline

5 years ago · 65a1fadfca
parent 0d0ebee9c0
commit 65a1fadfca
2 changed files with 4 additions and 3 deletions
--- 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
--- 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