bug fix: generated sample structure

5 years ago · 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)))