ftshijt commited on
Commit
9eec195
1 Parent(s): a8dd915

Update src/populate.py

Browse files
Files changed (1) hide show
  1. 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="max", numeric_only=True, ascending=True)
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