Update src/populate.py
Browse files- src/populate.py +4 -1
src/populate.py
CHANGED
@@ -29,8 +29,11 @@ def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchm
|
|
29 |
)
|
30 |
|
31 |
|
32 |
-
mos_rank = df[AutoEvalColumn.task2.name].rank(method="
|
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
|
|
|
29 |
)
|
30 |
|
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
|