DOC make a doctest more stable in MultiOutputClassifier
@@ -395,11 +395,12 @@ class MultiOutputClassifier(ClassifierMixin, _MultiOutputEstimator):
>>> import numpy as np
>>> from sklearn.datasets import make_multilabel_classification
>>> from sklearn.multioutput import MultiOutputClassifier
- >>> from sklearn.neighbors import KNeighborsClassifier
+ >>> from sklearn.linear_model import LogisticRegression
>>> X, y = make_multilabel_classification(n_classes=3, random_state=0)
- >>> clf = MultiOutputClassifier(KNeighborsClassifier()).fit(X, y)
+ >>> clf = MultiOutputClassifier(LogisticRegression()).fit(X, y)
>>> clf.predict(X[-2:])
- array([[1, 1, 0], [1, 1, 1]])
+ array([[1, 1, 1],
+ [1, 0, 1]])
"""
def __init__(self, estimator, *, n_jobs=None):
