Add train-split robust scaling for continuous values

2026-08-01 11:37:21 +08:00
parent 89dcf4b362
commit dfb22adf2d
6 changed files with 419 additions and 5 deletions
--- a/eval_data.py
+++ b/eval_data.py
@@ -161,6 +161,9 @@ def build_model_from_dataset(
        n_categories=dataset.n_categories,
        cont_type_ids=dataset.cont_type_ids,
        n_bins=int(cfg_get(args, cfg, "n_bins", 16)),
+        continuous_value_scaling=str(
+            cfg_get(args, cfg, "continuous_value_scaling", "none")
+        ),
        extra_pool_reduce=str(
            cfg_get(args, cfg, "extra_pool_reduce", "mean")
        ),