bug fix: generated sample structure

2020-02-25 12:33:44 -06:00 · 2020-02-25 12:33:44 -06:00 · 1cfd2059a4
parent 0656474ca8
commit 1cfd2059a4
1 changed files with 1 additions and 0 deletions
--- a/data/gan.py
+++ b/data/gan.py
@ -625,6 +625,7 @@ class Predict(GNet):
                
                tf.compat.v1.reset_default_graph()
                df = pd.DataFrame(df)
+                df.columns = columns
                return df.to_dict(orient='list')
                        # return df.to_dict(orient='list')
                        # count = str(len(os.listdir(self.out_dir)))