From 1cfd2059a472b59d6780e71d680c4ce7c4dbf0db Mon Sep 17 00:00:00 2001 From: Steve Nyemba Date: Tue, 25 Feb 2020 12:33:44 -0600 Subject: [PATCH] bug fix: generated sample structure --- data/gan.py | 1 + 1 file changed, 1 insertion(+) diff --git a/data/gan.py b/data/gan.py index f2f3cdb..d1b3123 100644 --- 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)))