Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -58,8 +58,8 @@ def load_leaderboard():
|
|
58 |
result['infer_tflop_s'] = result['infer_samples_per_sec'] * result['infer_gmacs'] * 2 / 1000
|
59 |
|
60 |
# Calculate average scores
|
61 |
-
top1_columns = [col for col in result.columns if col.endswith('_top1')]
|
62 |
-
top5_columns = [col for col in result.columns if col.endswith('_top5')]
|
63 |
result['avg_top1'] = result[top1_columns].mean(axis=1)
|
64 |
result['avg_top5'] = result[top5_columns].mean(axis=1)
|
65 |
|
|
|
58 |
result['infer_tflop_s'] = result['infer_samples_per_sec'] * result['infer_gmacs'] * 2 / 1000
|
59 |
|
60 |
# Calculate average scores
|
61 |
+
top1_columns = [col for col in result.columns if col.endswith('_top1') and not col == 'a_top1']
|
62 |
+
top5_columns = [col for col in result.columns if col.endswith('_top5') and not col == 'a_top5']
|
63 |
result['avg_top1'] = result[top1_columns].mean(axis=1)
|
64 |
result['avg_top5'] = result[top5_columns].mean(axis=1)
|
65 |
|