Update src/populate.py
Browse files- src/populate.py +3 -1
src/populate.py
CHANGED
|
@@ -32,9 +32,11 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
|
|
| 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
|
|
|
|
| 36 |
df["Rank"] = df.groupby("Precision").cumcount() + 1
|
| 37 |
df.pop("Ranking")
|
|
|
|
| 38 |
|
| 39 |
|
| 40 |
df = df[cols].round(decimals=2)
|
|
|
|
| 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["revert_task2"] = -df[AutoEvalColumn.task2.name]
|
| 36 |
+
df = df.sort_values(by=["Ranking", "revert_task2"], ascending=True)
|
| 37 |
df["Rank"] = df.groupby("Precision").cumcount() + 1
|
| 38 |
df.pop("Ranking")
|
| 39 |
+
df.pop("revert_task2")
|
| 40 |
|
| 41 |
|
| 42 |
df = df[cols].round(decimals=2)
|