Make it possible to pass an arbitrary probablistic classifier as method
for CalibratedClassifierCV
#21280
Labels
method
for CalibratedClassifierCV
#21280
Describe the workflow you want to enable
In addition to
method="sigmoid"
andmethod="isotonic"
it would be great to pass any scikit-learn compatible classifier with apredict_proba
toCalibratedClassifierCV
.In particular I would like to be able to pass:
method=HistGradientBoostingClassifier(monotonic_cst=[1])
(or #13649) to calibrate a model using a non-parametric method (similar to isotonic regression) but with an adjustable overfitting/underfitting tradeoff by settingmax_leaf_nodes
andmax_iter
(which is not possible with isotonic calibration).Describe your proposed solution
The
sklearn.calibration._fit_calibrator
private helper would need to accept and clone scikit-learn regressors instead of raisingValueError
.Of course one would need to add new tests + extend at least one of the calibration example to demonstrate this new capability.
Edit: the first version of this issue used
HistGradientBoostingRegressor
instead ofHistGradientBoostingClassifier
The text was updated successfully, but these errors were encountered: