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": {