Skip to content

Commit 30bd174

Browse files
committed
Improve csv table result processing for better sort when updating
1 parent e987e29 commit 30bd174

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

results/generate_csv_results.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,13 +62,13 @@ def diff(base_df, test_csv):
6262
test_df['rank_diff'] = rank_diff
6363

6464
test_df['param_count'] = test_df['param_count'].map('{:,.2f}'.format)
65-
test_df.sort_values('top1', ascending=False, inplace=True)
65+
test_df.sort_values(['top1', 'top5', 'model'], ascending=[False, False, True], inplace=True)
6666
test_df.to_csv(test_csv, index=False, float_format='%.3f')
6767

6868

6969
for base_results, test_results in results.items():
7070
base_df = pd.read_csv(base_results)
71-
base_df.sort_values('top1', ascending=False, inplace=True)
71+
base_df.sort_values(['top1', 'top5', 'model'], ascending=[False, False, True], inplace=True)
7272
for test_csv in test_results:
7373
diff(base_df, test_csv)
7474
base_df['param_count'] = base_df['param_count'].map('{:,.2f}'.format)

0 commit comments

Comments
 (0)