Update src/populate.py
Browse files- src/populate.py +0 -4
src/populate.py
CHANGED
@@ -31,15 +31,11 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
|
|
31 |
|
32 |
mos_rank = df[AutoEvalColumn.task2.name].rank(method="min", numeric_only=True, ascending=False)
|
33 |
bitrate_rank = df[AutoEvalColumn.task3.name].rank(method="min", numeric_only=True, ascending=True)
|
34 |
-
print(df[AutoEvalColumn.task2.name])
|
35 |
-
print(mos_rank)
|
36 |
-
print(bitrate_rank)
|
37 |
df["Ranking"] = pd.Series((mos_rank + bitrate_rank)/2)
|
38 |
df = df.sort_values(by=["Ranking", AutoEvalColumn.task2.name], ascending=True)
|
39 |
df["Rank"] = df.groupby("Precision").cumcount() + 1
|
40 |
df.pop("Ranking")
|
41 |
|
42 |
-
print(df)
|
43 |
|
44 |
df = df[cols].round(decimals=2)
|
45 |
|
|
|
31 |
|
32 |
mos_rank = df[AutoEvalColumn.task2.name].rank(method="min", numeric_only=True, ascending=False)
|
33 |
bitrate_rank = df[AutoEvalColumn.task3.name].rank(method="min", numeric_only=True, ascending=True)
|
|
|
|
|
|
|
34 |
df["Ranking"] = pd.Series((mos_rank + bitrate_rank)/2)
|
35 |
df = df.sort_values(by=["Ranking", AutoEvalColumn.task2.name], ascending=True)
|
36 |
df["Rank"] = df.groupby("Precision").cumcount() + 1
|
37 |
df.pop("Ranking")
|
38 |
|
|
|
39 |
|
40 |
df = df[cols].round(decimals=2)
|
41 |
|