update scripts

src/leaderboard/load_results.py

@@ -18,24 +18,22 @@ def load_data(data_path):
     for df_tmp in [df_m3exam, df_mmlu, df_avg]:
         df_tmp[['en', 'zh', 'id', 'th', 'vi', 'avg', 'avg_sea']] *= 100
         df_tmp[['en', 'zh', 'id', 'th', 'vi', 'avg', 'avg_sea']] = df_tmp[['en', 'zh', 'id', 'th', 'vi', 'avg', 'avg_sea']].round(2)
-        df_tmp['rank'] = df_tmp['avg'].rank(ascending=False).astype(int)
-
-        df_tmp = df_tmp[columns_sorted].sort_values(by='avg_sea', ascending=False)
-        df_tmp = df_tmp.rename(columns={'avg_sea': 'avg_sea⬆️'})
-        df_tmp['type'] = df_tmp['type'].map({'base': '🟢', 'chat': '🔶'})
-
-    # df_m3exam = df_m3exam[columns_sorted].sort_values(by='avg_sea', ascending=False)
-    # df_mmlu = df_mmlu[columns_sorted].sort_values(by='avg', ascending=False)
-    # df_avg = df_avg[columns_sorted].sort_values(by='avg', ascending=False)
-
-    # # change the column name from 'avg' to 'avg⬆️'
-    # df_m3exam = df_m3exam.rename(columns={'avg': 'avg⬆️'})
-    # df_mmlu = df_mmlu.rename(columns={'avg': 'avg⬆️'})
-    # df_avg = df_avg.rename(columns={'avg': 'avg⬆️'})
-
-    # # map the values in the 'type' column to the following values: {'base': 'Base', 'chat': 'Chat'}
-    # df_m3exam['type'] = df_m3exam['type'].map({'base': '🟢', 'chat': '🔶'})
-    # df_mmlu['type'] = df_mmlu['type'].map({'base': '🟢', 'chat': '🔶'})
-    # df_avg['type'] = df_avg['type'].map({'base': '🟢', 'chat': '🔶'})
+        df_tmp['rank'] = df_tmp['avg_sea'].rank(ascending=False).astype(int)
+        df_tmp = df_tmp[columns_sorted]
+
+    # sort the DataFrames by the 'avg_sea' column in descending order
+    df_m3exam = df_m3exam.sort_values(by='avg_sea', ascending=False)
+    df_mmlu = df_mmlu.sort_values(by='avg_sea', ascending=False)
+    df_avg = df_avg.sort_values(by='avg_sea', ascending=False)
+
+    # change the column name from 'avg_sea' to 'avg_sea⬆️'
+    df_m3exam = df_m3exam.rename(columns={'avg_sea': 'avg_sea⬆️'})
+    df_mmlu = df_mmlu.rename(columns={'avg_sea': 'avg_sea⬆️'})
+    df_avg = df_avg.rename(columns={'avg_sea': 'avg_sea⬆️'})
+
+    # map the values in the 'type' column to the following values: {'base': 'Base', 'chat': 'Chat'}
+    df_m3exam['type'] = df_m3exam['type'].map({'base': '🟢', 'chat': '🔶'})
+    df_mmlu['type'] = df_mmlu['type'].map({'base': '🟢', 'chat': '🔶'})
+    df_avg['type'] = df_avg['type'].map({'base': '🟢', 'chat': '🔶'})
 
     return df_m3exam, df_mmlu, df_avg