Some fixes due to changes in data format

scikit-learn · Aug 15, 2018 · df22dfa · df22dfa
1 parent 22e1ab8
commit df22dfa
Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/examples/linear_model/plot_sgd_early_stopping.py b/examples/linear_model/plot_sgd_early_stopping.py
@@ -56,7 +56,7 @@
 print(__doc__)
 
 
-def load_mnist(n_samples=None, class_0=0, class_1=8):
+def load_mnist(n_samples=None, class_0='0', class_1='8'):
     """Load MNIST, select two classes, shuffle and return only n_samples."""
     mnist = fetch_openml('mnist_784', version=1)
 

diff --git a/examples/multioutput/plot_classifier_chain_yeast.py b/examples/multioutput/plot_classifier_chain_yeast.py
@@ -47,9 +47,10 @@
 print(__doc__)
 
 # Load a multi-label dataset
-yeast = fetch_openml('yeast', version=2)
-X_train, X_test, Y_train, Y_test = train_test_split(yeast.data, yeast.target,
-                                                    test_size=.2,
+yeast = fetch_openml('yeast', version=4)
+X = yeast.data
+Y = yeast.target == 'TRUE'
+X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=.2,
                                                     random_state=0)
 
 # Fit an independent logistic regression model for each class using the