Skip to content

Commit ee19b40

Browse files
committed
use private function
1 parent 535353b commit ee19b40

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

stlearn/stacking.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,7 @@ def predict_proba(self, X):
215215
return _predict_proba_estimator(self.stacking_estimator, predictions_)
216216

217217
def decision_function(self, X):
218-
return self.predict_proba(X)[:, -1]
218+
return self._disambiguate_probability(self.predict_proba(X))
219219

220220
def score(self, X, y):
221221
"""Returns the mean accuracy on the given test data and labels.

0 commit comments

Comments
 (0)