8000 FIX AdaBoost ZeroDivisionError in proba #7501 - tests corrected · scikit-learn/scikit-learn@9a386a2 · GitHub
[go: up one dir, main page]

Skip to content

Commit 9a386a2

Browse files
dokatoDominik Krzeminski
authored andcommitted
FIX AdaBoost ZeroDivisionError in proba #7501 - tests corrected
1 parent 1816bc5 commit 9a386a2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/ensemble/tests/test_weight_boosting.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ def test_oneclass_proba():
7878
# Test `predict_proba` robustness for one class label input.< 5F1E /span>
7979
y_t = np.ones((len(X),))
8080
clf = AdaBoostClassifier().fit(X, y_t)
81-
assert_array_equal(clf.predict_proba(X), np.ones((X.shape[0], 1)))
81+
assert_array_equal(clf.predict_proba(X), np.ones((len(X), 1)))
8282

8383

8484
def test_classification_toy():

0 commit comments

Comments
 (0)
0