Add display_names to all methods

Author: Innixma

tabarena/tabarena/nips2025_utils/artifacts/_tabarena_method_metadata_2025_06_12.py

@@ -95,6 +95,32 @@
     "TabPFNv2_GPU": "TABPFNV2",
 }
 
+
+methods_display_name_map = {
+    "CatBoost": "CatBoost",
+    "Dummy": "Dummy",
+    "ExplainableBM": "EBM",
+    "ExtraTrees": "ExtraTrees",
+    "KNeighbors": "KNN",
+    "LightGBM": "LightGBM",
+    "LinearModel": "Linear",
+    "ModernNCA": "ModernNCA (CPU)",
+    "NeuralNetFastAI": "FastaiMLP",
+    "NeuralNetTorch": "TorchMLP",
+    "RandomForest": "RandomForest",
+    "RealMLP": "RealMLP (CPU)",
+    "TabM": "TabM (CPU)",
+    "XGBoost": "XGBoost",
+
+    "ModernNCA_GPU": "ModernNCA",
+    "RealMLP_GPU": "RealMLP",
+    "TabDPT_GPU": "TabDPT",
+    "TabICL_GPU": "TabICL",
+    "TabM_GPU": "TabM",
+    "TabPFNv2_GPU": "TabPFNv2",
+}
+
+
 methods_config_default_map = {
     "CatBoost": "CatBoost_c1_BAG_L1",
     "Dummy": "Dummy_c1_BAG_L1",
@@ -243,6 +269,7 @@
     config_default = methods_config_default_map[method]
     is_bag = method in methods_is_bag
     upload_as_public = method not in methods_upload_as_private
+    display_name = methods_display_name_map.get(method, None)
     assert compute_type in ["cpu", "gpu"]
     if compute_type == "cpu":
         method_kwargs = cpu_kwargs
@@ -260,6 +287,7 @@
     method_metadata = MethodMetadata(
         method=method,
         config_default=config_default,
+        display_name=display_name,
         ag_key=ag_key,
         is_bag=is_bag,
         can_hpo=can_hpo,
@@ -273,6 +301,7 @@
 
 ag_130_metadata = MethodMetadata(
     method="AutoGluon_v130",
+    name="AutoGluon 1.3 (best, 4h)",
     artifact_name="tabarena-2025-06-12",
     date="2025-06-12",
     method_type="baseline",

tabarena/tabarena/nips2025_utils/artifacts/_tabarena_method_metadata_2025_09_03.py

@@ -17,6 +17,7 @@
 ag_140_metadata = MethodMetadata(
     method="AutoGluon_v140",
     method_type="baseline",
+    display_name="AutoGluon 1.4 (4h)",
     compute="gpu",
     date="2025-09-03",
     verified=True,
@@ -26,6 +27,7 @@
 mitra_metadata = MethodMetadata(
     method="Mitra_GPU",
     method_type="config",
+    display_name="Mitra",
     compute="gpu",
     date="2025-09-03",
     ag_key="MITRA",
@@ -40,6 +42,7 @@
 limix_metadata = MethodMetadata(
     method="LimiX_GPU",
     method_type="config",
+    display_name="LimiX",
     compute="gpu",
     date="2025-09-03",
     ag_key="LIMIX",
@@ -54,6 +57,7 @@
 realmlp_gpu_metadata = MethodMetadata(
     method="RealMLP_GPU",
     method_type="config",
+    display_name="RealMLP",
     compute="gpu",
     date="2025-09-03",
     ag_key="TA-REALMLP",
@@ -68,6 +72,7 @@
 ebm_metadata = MethodMetadata(
     method="ExplainableBM",
     method_type="config",
+    display_name="EBM",
     compute="cpu",
     date="2025-09-03",
     ag_key="EBM",
@@ -81,6 +86,7 @@
 xrfm_metadata = MethodMetadata(
     method="xRFM_GPU",
     method_type="config",
+    display_name="xRFM",
     compute="gpu",
     date="2025-09-03",
     ag_key="XRFM",
@@ -95,6 +101,7 @@
 tabflex_metadata = MethodMetadata(
     method="TabFlex_GPU",
     method_type="config",
+    display_name="TabFlex",
     compute="gpu",
     date="2025-09-03",
     ag_key="TABFLEX",
@@ -109,6 +116,7 @@
 betatabpfn_metadata = MethodMetadata(
     method="BetaTabPFN_GPU",
     method_type="config",
+    display_name="BetaTabPFN",
     compute="gpu",
     date="2025-09-03",
     ag_key="BETA",

tabarena/tabarena/nips2025_utils/artifacts/_tabarena_method_metadata_2025_10_20.py

@@ -16,6 +16,7 @@
 lr_metadata = MethodMetadata(
     method="LinearModel",
     method_type="config",
+    display_name="Linear",
     compute="cpu",
     date="2025-10-20",
     ag_key="LR",
@@ -30,6 +31,7 @@
 knn_metadata = MethodMetadata(
     method="KNeighbors",
     method_type="config",
+    display_name="KNN",
     compute="cpu",
     date="2025-10-20",
     ag_key="KNN",
@@ -44,6 +46,7 @@
 tabdpt_metadata = MethodMetadata(
     method="TabDPT_GPU",
     method_type="config",
+    display_name="TabDPT",
     compute="gpu",
     date="2025-10-20",
     ag_key="TABDPT",

tabarena/tabarena/nips2025_utils/artifacts/_tabarena_method_metadata_2025_11_12.py

@@ -3,6 +3,7 @@
 realtabpfn25_metadata = MethodMetadata(
     method="RealTabPFN-v2.5",
     method_type="config",
+    display_name="RealTabPFN-2.5",
     compute="gpu",
     date="2025-11-12",
     ag_key="REALTABPFN-V2.5",
@@ -25,6 +26,7 @@
 contexttab_metadata = MethodMetadata(
     method="SAP-RPT-OSS",
     method_type="config",
+    display_name="SAP-RPT-OSS",
     compute="gpu",
     date="2025-11-25",
     ag_key="SAP-RPT-OSS",