search

arogozhnikov / hep_ml

Machine Learning for High Energy Physics.
https://arogozhnikov.github.io/hep_ml/
Other
176 stars 64 forks source link

Upgrade sklearn criterion mse to squared_error #71

Closed jonas-eschle closed 2 years ago

jonas-eschle commented 2 years ago

mse is deprecated and will be removed: https://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeRegressor.html

squared_error is equivalent.

jonas-eschle commented 2 years ago

Also require sklearn to be >= 1 (that's where squared_error is used)

Ready from my side

arogozhnikov commented 2 years ago

Thanks a lot for taking time to update, Jonas