From 17e83cb219dcd172f3ed3b61701d2117af2e08ab Mon Sep 17 00:00:00 2001 From: Eden Wu Date: Sun, 1 Sep 2024 03:50:12 -0400 Subject: [PATCH] debug --- alpha_automl/hyperparameter_tuning/smac_parameters.json | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/alpha_automl/hyperparameter_tuning/smac_parameters.json b/alpha_automl/hyperparameter_tuning/smac_parameters.json index 8c2c968..06559d4 100644 --- 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": {