From bad1e669aee4f1127efd78a94e1c6c4bb4b9b233 Mon Sep 17 00:00:00 2001 From: Jingang Qu Date: Sat, 26 Sep 2026 01:56:08 -0700 Subject: [PATCH] Use 8 estimators for small and medium `KumoTabular` in benchmarks The size options from #983 set 16 estimators for every size; large keeps 16. Signed-off-by: Jingang Qu --- benchmark/tabular/model.py | 2 ++ benchmark/tabular/scoringbench/models.py | 4 ++-- benchmark/tabular/talent/models.py | 4 ++-- 3 files changed, 6 insertions(+), 4 deletions(-) diff --git a/benchmark/tabular/model.py b/benchmark/tabular/model.py index c1decf7f7..f363ae972 100644 --- a/benchmark/tabular/model.py +++ b/benchmark/tabular/model.py @@ -404,12 +404,14 @@ class SDMKumoTabularSmallModel(SDMKumoTabularModel): ag_key = "SDM-KUMO-TABULAR-SMALL" ag_name = "SDMKumoTabularSmall" size = "small" + default_num_estimators = 8 class SDMKumoTabularMediumModel(SDMKumoTabularModel): ag_key = "SDM-KUMO-TABULAR-MEDIUM" ag_name = "SDMKumoTabularMedium" size = "medium" + default_num_estimators = 8 class SDMKumoTabularLargeModel(SDMKumoTabularModel): diff --git a/benchmark/tabular/scoringbench/models.py b/benchmark/tabular/scoringbench/models.py index a3b00be0d..98c932191 100644 --- a/benchmark/tabular/scoringbench/models.py +++ b/benchmark/tabular/scoringbench/models.py @@ -56,14 +56,14 @@ def _create_kumo_tabular( method="sdm_kumo_tabular_small", factory=partial(_create_kumo_tabular, size="small"), autocast_dtype=torch.float16, - num_estimators=16, + num_estimators=8, ), "kumo-tabular-medium": ModelConfig( name="KumoTabular-Medium", method="sdm_kumo_tabular_medium", factory=partial(_create_kumo_tabular, size="medium"), autocast_dtype=torch.float16, - num_estimators=16, + num_estimators=8, ), "kumo-tabular-large": ModelConfig( name="KumoTabular-Large", diff --git a/benchmark/tabular/talent/models.py b/benchmark/tabular/talent/models.py index 94ab858f7..35350a763 100644 --- a/benchmark/tabular/talent/models.py +++ b/benchmark/tabular/talent/models.py @@ -86,14 +86,14 @@ def _create_tabfm( "kumo-tabular-small": ModelConfig( name="KumoTabular-Small", factory=partial(_create_kumo_tabular, size="small"), - num_estimators=16, + num_estimators=8, autocast_dtype=torch.float16, max_classes=10, ), "kumo-tabular-medium": ModelConfig( name="KumoTabular-Medium", factory=partial(_create_kumo_tabular, size="medium"), - num_estimators=16, + num_estimators=8, autocast_dtype=torch.float16, max_classes=10, ),