bug fix: data-frame should be returned (not series)

5 years ago · 0656474ca8
parent 94780281d0
commit 0656474ca8
1 changed files with 1 additions and 1 deletions
--- a/data/gan.py
+++ b/data/gan.py
@ -624,7 +624,7 @@ class Predict(GNet):
                        
                
                tf.compat.v1.reset_default_graph()
-                
+                df = pd.DataFrame(df)
                return df.to_dict(orient='list')
                        # return df.to_dict(orient='list')
                        # count = str(len(os.listdir(self.out_dir)))