diff --git a/sklearn/ensemble/tests/test_gradient_boosting.py b/sklearn/ensemble/tests/test_gradient_boosting.py index 3ae437add6720ce8f379992623f5e99c4a1d532e..9f49024813878a56cc95c1cf67c720a2f82d04a9 100644 --- a/sklearn/ensemble/tests/test_gradient_boosting.py +++ b/sklearn/ensemble/tests/test_gradient_boosting.py @@ -214,7 +214,7 @@ def test_probability(): assert np.all(y_proba <= 1.0) # derive predictions from probabilities - y_pred = clf.classes.take(y_proba.argmax(axis=1), axis=0) + y_pred = clf.classes_.take(y_proba.argmax(axis=1), axis=0) assert_array_equal(y_pred, true_result)