Closed fkiraly closed 6 months ago
You are right that sklearn.tree.DecisionTreeRegressor
does support "absolute_error" and "poisson", however the gradient boosting models in sklearn do not, e.g. sklearn.ensemble.GradientBoostingClassifier
. Not sure why, but I would stick to the same options that gradient boosting models in sklearn support.
In
GradientBoostingSurvivalAnalysis
, docstring and logic do not agree on the set of possiblecriterion
values.Docstring implies that
"friedman_mse"
,"mse"
, and"mae"
are valid, but passing"mse"
or"mae"
triggers aValueError
which says that only"friedman_mse"
and"squared_error"
are valid.Indeed, the constraints from
BaseGradientBoosting
have"criterion": [StrOptions({"friedman_mse", "squared_error"})],
.This seems consistent with the construction of
DecisionTreeRegressor
under the hood (which would allow additional options, though).