Upload

src/structures/ldek_structure.py

@@ -21,7 +21,7 @@ def generate_ORDER_LIST_LDEK_and_data_types(json_data):
     data_types = ["markdown", "number"]
 
     for key in json_data.keys():
-        if key not in ["model_name"]:
+        if key not in ["model_name", "overall_accuracy"]:
             ORDER_LIST_LDEK.append(key)
             data_types.append("number")
     ORDER_LIST_LDEK[2:] = sorted(ORDER_LIST_LDEK[2:])
@@ -32,26 +32,38 @@ def filter_columns_ldek(column_choices):
     return LDEK_ACCS[selected_columns]
 
 def load_json_data(file_path, ORDER_LIST_LDEK):
+    # Load JSON data into a DataFrame
     LDEK_ACCS = pd.read_json(file_path)
+
+    # Convert nested dictionary fields (if any) to strings
     for column in LDEK_ACCS.columns:
         if LDEK_ACCS[column].apply(type).eq(dict).any():
             LDEK_ACCS[column] = LDEK_ACCS[column].apply(str)
 
+    # Apply the custom model name replacement function
     LDEK_ACCS["model_name"] = LDEK_ACCS["model_name"].apply(
         lambda name: replace_models_names(name)
     )
 
+    # Convert numeric columns and round
     for column in LDEK_ACCS.select_dtypes(include='number').columns:
-            LDEK_ACCS[column] = LDEK_ACCS[column].round(2)
+        LDEK_ACCS[column] = LDEK_ACCS[column].round(2)
+
+    # Ensure overall_accuracy is treated as numeric for sorting
+    LDEK_ACCS["overall_accuracy"] = pd.to_numeric(LDEK_ACCS["overall_accuracy"], errors='coerce')
+
+    # Reorder columns based on ORDER_LIST_LDEK
     ordered_columns = [col for col in ORDER_LIST_LDEK if col in LDEK_ACCS.columns]
     LDEK_ACCS = LDEK_ACCS[ordered_columns]
 
-    if "Score" in LDEK_ACCS.columns:
-        LDEK_ACCS = LDEK_ACCS.sort_values(by="Score", ascending=False)
+    # Sort by overall_accuracy in descending order if it exists
+    if "overall_accuracy" in LDEK_ACCS.columns:
+        LDEK_ACCS = LDEK_ACCS.sort_values(by="overall_accuracy", ascending=False)
 
     return LDEK_ACCS
 
 
+
 # file_path = str(abs_path / "leaderboards/r_ldek_report_scores.json")
 file_path = str(abs_path / "leaderboards/ldek_accs.json")
 with open(file_path, 'r', encoding='utf-8') as file: