debug

VIDA-NYU · Sep 1, 2024 · 17e83cb · 17e83cb
1 parent 1da9766
commit 17e83cb
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/alpha_automl/hyperparameter_tuning/smac_parameters.json b/alpha_automl/hyperparameter_tuning/smac_parameters.json
@@ -334,10 +334,12 @@
         "criterion": {
             "type": "Categorical",
             "value": [
-                "entropy",
-                "gini"
+                "squared_error",
+                "absolute_error",
+                "friedman_mse",
+                "poisson"
             ],
-            "default": "entropy"
+            "default": "squared_error"
         }
     },
     "xgboost.XGBRegressor": {