Improve csv table result processing for better sort when updating

pull/1606/head
Ross Wightman 2 years ago
parent e987e29036
commit 30bd1746c5

@ -62,13 +62,13 @@ def diff(base_df, test_csv):
test_df['rank_diff'] = rank_diff
test_df['param_count'] = test_df['param_count'].map('{:,.2f}'.format)
test_df.sort_values('top1', ascending=False, inplace=True)
test_df.sort_values(['top1', 'top5', 'model'], ascending=[False, False, True], inplace=True)
test_df.to_csv(test_csv, index=False, float_format='%.3f')
for base_results, test_results in results.items():
base_df = pd.read_csv(base_results)
base_df.sort_values('top1', ascending=False, inplace=True)
base_df.sort_values(['top1', 'top5', 'model'], ascending=[False, False, True], inplace=True)
for test_csv in test_results:
diff(base_df, test_csv)
base_df['param_count'] = base_df['param_count'].map('{:,.2f}'.format)

Loading…
Cancel
Save