bug fix: generated sample structure

This commit is contained in:
Steve Nyemba 2020-02-25 12:33:44 -06:00
parent 0656474ca8
commit 1cfd2059a4
1 changed files with 1 additions and 0 deletions

View File

@ -625,6 +625,7 @@ class Predict(GNet):
tf.compat.v1.reset_default_graph() tf.compat.v1.reset_default_graph()
df = pd.DataFrame(df) df = pd.DataFrame(df)
df.columns = columns
return df.to_dict(orient='list') return df.to_dict(orient='list')
# return df.to_dict(orient='list') # return df.to_dict(orient='list')
# count = str(len(os.listdir(self.out_dir))) # count = str(len(os.listdir(self.out_dir)))