bug fix: generated sample structure
This commit is contained in:
parent
0656474ca8
commit
1cfd2059a4
|
@ -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)))
|
||||||
|
|
Loading…
Reference in New Issue