tangtang commited on
Commit
6546def
·
1 Parent(s): bd8143b

Update space1

Browse files
Files changed (1) hide show
  1. src/populate.py +7 -4
src/populate.py CHANGED
@@ -2,7 +2,7 @@ import json
2
  import os
3
 
4
  import pandas as pd
5
-
6
  from src.display.formatting import has_no_nan_values, make_clickable_model
7
  from src.display.utils import AutoEvalColumn, EvalQueueColumn
8
  from src.leaderboard.read_evals import get_raw_eval_results
@@ -15,13 +15,16 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
15
  print(all_data_json)
16
  df = pd.DataFrame.from_records(all_data_json)
17
  print(df)
 
 
18
  df = df.sort_values(by=["Precision (%)"], ascending=False)
19
 
 
20
  # 假设用 Precision (%) 和 Title search rate (%) 的平均值
21
- df["Average ⬆️"] = df[["Precision (%)", "Title search rate (%)"]].mean(axis=1)
22
 
23
- # 然后排序
24
- df = df.sort_values(by=["Average ⬆️"], ascending=False)
25
 
26
  # 再保留需要显示的列
27
  cols = [c for c in cols if c in df.columns]
 
2
  import os
3
 
4
  import pandas as pd
5
+ import numpy as np
6
  from src.display.formatting import has_no_nan_values, make_clickable_model
7
  from src.display.utils import AutoEvalColumn, EvalQueueColumn
8
  from src.leaderboard.read_evals import get_raw_eval_results
 
15
  print(all_data_json)
16
  df = pd.DataFrame.from_records(all_data_json)
17
  print(df)
18
+ df["Precision (%)"] = df["Precision (%)"].apply(lambda x: x[0] if len(x) > 0 else np.nan)
19
+ df["Title search rate (%)"] = df["Title search rate (%)"].apply(lambda x: x[0] if len(x) > 0 else np.nan)
20
  df = df.sort_values(by=["Precision (%)"], ascending=False)
21
 
22
+
23
  # 假设用 Precision (%) 和 Title search rate (%) 的平均值
24
+ df["Average"] = df[["Precision (%)", "Title search rate (%)"]].mean(axis=1)
25
 
26
+ # # 然后排序
27
+ df = df.sort_values(by=["Average"], ascending=False)
28
 
29
  # 再保留需要显示的列
30
  cols = [c for c in cols if c in df.columns]