Skip to content

Commit 0ca4ac2

Browse files
MNT Use float64 epsilon when clipping initial probabilities in GradientBoosting (scikit-learn#31575)
Co-authored-by: Jérémie du Boisberranger <jeremie@probabl.ai>
1 parent e15920c commit 0ca4ac2

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

sklearn/ensemble/_gb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def _init_raw_predictions(X, estimator, loss, use_predict_proba):
114114
predictions = estimator.predict_proba(X)
115115
if not loss.is_multiclass:
116116
predictions = predictions[:, 1] # probability of positive class
117-
eps = np.finfo(np.float32).eps # FIXME: This is quite large!
117+
eps = np.finfo(np.float64).eps
118118
predictions = np.clip(predictions, eps, 1 - eps, dtype=np.float64)
119119
else:
120120
predictions = estimator.predict(X).astype(np.float64)

0 commit comments

Comments
 (0)