jasonshaoshun commited on
Commit
015ea11
·
1 Parent(s): 447b70d
Files changed (1) hide show
  1. app.py +17 -3
app.py CHANGED
@@ -360,11 +360,27 @@ def init_leaderboard_mib_subgraph(dataframe, track):
360
 
361
  # First, create our display name mapping
362
  # This is like creating a translation dictionary between internal names and display names
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
363
  display_mapping = {}
364
  for task in TasksMib_Subgraph:
365
  for model in task.value.models:
366
  field_name = f"{task.value.benchmark}_{model}"
367
- display_name = f"{task.value.benchmark}({model})"
368
  display_mapping[field_name] = display_name
369
 
370
  # Now when creating benchmark groups, we'll use display names
@@ -402,8 +418,6 @@ def init_leaderboard_mib_subgraph(dataframe, track):
402
  # Important: We need to rename our DataFrame columns to match display names
403
 
404
  renamed_df = dataframe.rename(columns=display_mapping)
405
-
406
- renamed_df = dataframe
407
  all_columns = dataframe.columns.tolist()
408
 
409
  # Original code
 
360
 
361
  # First, create our display name mapping
362
  # This is like creating a translation dictionary between internal names and display names
363
+ model_name_mapping = {
364
+ "qwen2_5": "Qwen-2.5",
365
+ "gpt2": "GPT-2",
366
+ "gemma2": "Gemma-2",
367
+ "llama3": "LLaMA-3"
368
+ }
369
+
370
+ benchmark_mapping = {
371
+ "ioi": "IOI",
372
+ "mcqa": "MCQA",
373
+ "arithmetic_addition": "Arithmetic Addition",
374
+ "arithmetic_subtraction": "Arithmetic Subtraction",
375
+ "arc_easy": "ARC Easy",
376
+ "arc_challenge": "ARC Challenge"
377
+ }
378
+
379
  display_mapping = {}
380
  for task in TasksMib_Subgraph:
381
  for model in task.value.models:
382
  field_name = f"{task.value.benchmark}_{model}"
383
+ display_name = f"{benchmark_mapping[task.value.benchmark]}({model_name_mapping[model]})"
384
  display_mapping[field_name] = display_name
385
 
386
  # Now when creating benchmark groups, we'll use display names
 
418
  # Important: We need to rename our DataFrame columns to match display names
419
 
420
  renamed_df = dataframe.rename(columns=display_mapping)
 
 
421
  all_columns = dataframe.columns.tolist()
422
 
423
  # Original code