kennymckormick commited on
Commit
d7dd071
1 Parent(s): 7b3b94d
Files changed (1) hide show
  1. 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
- res[d].append(item[d]['Overall'])
80
- scores.append(item[d]['Overall'] / 28)
81
  elif d == 'OCRBench':
82
- res[d].append(item[d]['Final Score'])
83
- scores.append(item[d]['Final Score'] / 10)
84
  else:
85
- res[d].append(item[d]['Overall'])
86
- scores.append(item[d]['Overall'])
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