Skip to content

Commit ebb0645

Browse files
ogriseljnothman
authored andcommitted
DOC reference CalibratedClassifierCV in SGDClassifier.predict_proba (scikit-learn#7281)
1 parent 1e2b443 commit ebb0645

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

sklearn/linear_model/stochastic_gradient.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -734,7 +734,10 @@ def predict_proba(self):
734734
Elkan.
735735
736736
Binary probability estimates for loss="modified_huber" are given by
737-
(clip(decision_function(X), -1, 1) + 1) / 2.
737+
(clip(decision_function(X), -1, 1) + 1) / 2. For other loss functions
738+
it is necessary to perform proper probability calibration by wrapping
739+
the classifier with
740+
:class:`sklearn.calibration.CalibratedClassifierCV` instead.
738741
739742
Parameters
740743
----------

0 commit comments

Comments
 (0)