sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)
https://developers.sber.ru/portal/products/lightautoml
Apache License 2.0
1.08k stars 47 forks source link

RandomForestRegressor bug #134

Closed BELONOVSKII closed 8 months ago

BELONOVSKII commented 9 months ago

🐛 Bug

In the recent versions of sklearn ( >= 1.0.0) the criterion parameter of RandomForestRegressor has changed

python3.10/site-packages/sklearn/utils/_param_validation.py", line 95, in validate_parameter_constraints raise InvalidParameterError( sklearn.utils._param_validation.InvalidParameterError: The 'criterion' parameter of RandomForestRegressor must be a str among {'poisson', 'friedman_mse', 'squared_error', 'absolute_error'}. Got 'mse' instead.