s
This commit is contained in:
parent
083fa8a154
commit
bdaba18487
@ -33,7 +33,7 @@ if os.path.exists(metrics_file):
|
|||||||
metrics_df = pd.read_csv(metrics_file)
|
metrics_df = pd.read_csv(metrics_file)
|
||||||
else:
|
else:
|
||||||
metrics_df = pd.DataFrame(columns=['top_1_accuracy', 'top_5_accuracy'])
|
metrics_df = pd.DataFrame(columns=['top_1_accuracy', 'top_5_accuracy'])
|
||||||
metrics_df = metrics_df.append({'top_1_accuracy': np.mean(top_1_accuracy), 'top_5_accuracy': np.mean(top_5_accuracy)}, ignore_index=True)
|
metrics_df = metrics_df.concat({'top_1_accuracy': np.mean(top_1_accuracy), 'top_5_accuracy': np.mean(top_5_accuracy)}, ignore_index=True)
|
||||||
metrics_df.to_csv(metrics_file, index=False)
|
metrics_df.to_csv(metrics_file, index=False)
|
||||||
|
|
||||||
plt.figure(figsize=(10, 6))
|
plt.figure(figsize=(10, 6))
|
||||||
|
Loading…
Reference in New Issue
Block a user