Skip to content
This repository was archived by the owner on Oct 3, 2023. It is now read-only.

Commit a05bc8d

Browse files
committed
Merge pull request scikit-learn#5858 from mcculloh/gradientboost
[MRG+1] Change super call to pass presort value to super
2 parents bf00574 + 221e7ce commit a05bc8d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sklearn/ensemble/gradient_boosting.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1014,7 +1014,7 @@ def fit(self, X, y, sample_weight=None, monitor=None):
10141014
elif presort == 'auto':
10151015
presort = True
10161016

1017-
if self.presort == True:
1017+
if presort == True:
10181018
if issparse(X):
10191019
raise ValueError("Presorting is not supported for sparse matrices.")
10201020
else:
@@ -1787,7 +1787,7 @@ def __init__(self, loss='ls', learning_rate=0.1, n_estimators=100,
17871787
max_features=max_features,
17881788
random_state=random_state, alpha=alpha, verbose=verbose,
17891789
max_leaf_nodes=max_leaf_nodes, warm_start=warm_start,
1790-
presort='auto')
1790+
presort=presort)
17911791

17921792
def predict(self, X):
17931793
"""Predict regression target for X.

0 commit comments

Comments
 (0)