update readme example

johny-c · Nov 4, 2018 · a0857e9 · a0857e9
1 parent de5290c
commit a0857e9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/README.rst b/README.rst
@@ -60,8 +60,7 @@ Here is a minimal use case:
 
 
     # Load a data set
-    dataset = load_iris()
-    X, y = dataset.data, dataset.target
+    X, y = load_iris(return_X_y=True)
 
     # Split in training and testing set
     X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.7, stratify=y, random_state=42)

diff --git a/examples/minimal.py b/examples/minimal.py
@@ -6,8 +6,7 @@
 
 
 # Load a data set
-dataset = load_iris()
-X, y = dataset.data, dataset.target
+X, y = load_iris(return_X_y=True)
 
 # Split in training and testing set
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.7, stratify=y, random_state=42)