Update dataset names

src/populate.py

@@ -19,6 +19,18 @@ model_names = {
     'SDV': "SDV",
 }
 
+# Dataset name mapping dictionary
+dataset_names = {
+    "airbnb-simplified_subsampled": "Airbnb",
+    "Berka_subsampled": "Berka",
+    "Biodegradability_v1": "Biodegradability",
+    "CORA_v1": "Cora",
+    "imdb_MovieLens_v1": "IMDB",
+    "rossmann_subsampled": "Rossmann",
+    "walmart_subsampled": "Walmart",
+    "f1_subsampled": "F1",
+}
+
 
 # def get_leaderboard_df(results_path: str, requests_path: str, cols: list, benchmark_cols: list) -> pd.DataFrame:
 #     """Creates a dataframe from all the individual experiment results"""
@@ -76,7 +88,12 @@ def get_leaderboard_df(results_path: str, cols: list, benchmark_cols: list) -> p
         # Rename model if it exists in the mapping dictionary
         if model.upper() in model_names:
             model = model_names[model.upper()]
+        
         dataset = data["dataset_name"]
+        # Rename dataset if it exists in the mapping dictionary
+        if dataset in dataset_names:
+            dataset = dataset_names[dataset]
+            
         row = {"Dataset": dataset, "Model": model}
         for metric in multi_table_metrics:
             stripped_metric = strip_emoji(metric)