kexinhuang12345 commited on
Commit
139368d
·
1 Parent(s): 6d97820
Files changed (1) hide show
  1. src/populate.py +1 -0
src/populate.py CHANGED
@@ -52,6 +52,7 @@ def get_leaderboard_df(EVAL_REQUESTS_PATH, tasks) -> pd.DataFrame:
52
 
53
  columns_to_show = ['model', 'author', 'email', 'paper_url', 'github_url', 'submitted_time', 'params'] + list(name2short_name.values())
54
  df_res = pd.DataFrame([{col: model[col] for col in columns_to_show} for model in model_res])
 
55
  ranks = df_res[list(name2short_name.values())].rank()
56
  df_res.rename(columns={'model': 'Model', 'author': 'Author', 'email': 'Email', 'paper_url': 'Paper URL', 'github_url': 'Github URL', 'submitted_time': 'Time', 'params': '# of Params'}, inplace=True)
57
  df_res['Average Rank⬆️'] = ranks.mean(axis=1)
 
52
 
53
  columns_to_show = ['model', 'author', 'email', 'paper_url', 'github_url', 'submitted_time', 'params'] + list(name2short_name.values())
54
  df_res = pd.DataFrame([{col: model[col] for col in columns_to_show} for model in model_res])
55
+ print(df_res)
56
  ranks = df_res[list(name2short_name.values())].rank()
57
  df_res.rename(columns={'model': 'Model', 'author': 'Author', 'email': 'Email', 'paper_url': 'Paper URL', 'github_url': 'Github URL', 'submitted_time': 'Time', 'params': '# of Params'}, inplace=True)
58
  df_res['Average Rank⬆️'] = ranks.mean(axis=1)