kennymckormick
commited on
Commit
•
d7dd071
1
Parent(s):
7b3b94d
update
Browse files- gen_table.py +6 -8
gen_table.py
CHANGED
@@ -75,17 +75,15 @@ def BUILD_L1_DF(results, fields):
|
|
75 |
res[k].append(meta[k])
|
76 |
scores, ranks = [], []
|
77 |
for d in fields:
|
|
|
|
|
78 |
if d == 'MME':
|
79 |
-
|
80 |
-
scores.append(item[d]['Overall'] / 28)
|
81 |
elif d == 'OCRBench':
|
82 |
-
|
83 |
-
scores.append(item[d]['Final Score'] / 10)
|
84 |
else:
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
ranks.append(nth_large(item[d]['Overall'], [x[d]['Overall'] for x in results.values()]))
|
89 |
res['Avg Score'].append(round(np.mean(scores), 1))
|
90 |
res['Avg Rank'].append(round(np.mean(ranks), 2))
|
91 |
|
|
|
75 |
res[k].append(meta[k])
|
76 |
scores, ranks = [], []
|
77 |
for d in fields:
|
78 |
+
key_name = 'Overall' if d != 'OCRBench' else 'Final Score'
|
79 |
+
res[d].append(item[d][key_name])
|
80 |
if d == 'MME':
|
81 |
+
scores.append(item[d][key_name] / 28)
|
|
|
82 |
elif d == 'OCRBench':
|
83 |
+
scores.append(item[d][key_name] / 10)
|
|
|
84 |
else:
|
85 |
+
scores.append(item[d][key_name])
|
86 |
+
ranks.append(nth_large(item[d][key_name], [x[d][key_name] for x in results.values()]))
|
|
|
|
|
87 |
res['Avg Score'].append(round(np.mean(scores), 1))
|
88 |
res['Avg Rank'].append(round(np.mean(ranks), 2))
|
89 |
|