Spaces:
Runtime error
Runtime error
tangtang
commited on
Commit
·
6546def
1
Parent(s):
bd8143b
Update space1
Browse files- 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
|
| 22 |
|
| 23 |
-
# 然后排序
|
| 24 |
-
df = df.sort_values(by=["Average
|
| 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]
|