Fixes #22: Added cross_validation to iris.py and iris_custom_model.py

tensorflow · Dec 8, 2015 · 43cd5c0 · 43cd5c0
1 parent 9b17502
commit 43cd5c0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/iris.py b/examples/iris.py
@@ -14,7 +14,7 @@
 
 import random
 
-from sklearn import datasets, metrics
+from sklearn import datasets, metrics, cross_validation
 
 import skflow
 

diff --git a/examples/iris_custom_model.py b/examples/iris_custom_model.py
@@ -15,7 +15,7 @@
 import random
 
 import skflow
-from sklearn import datasets, metrics
+from sklearn import datasets, metrics, cross_validation
 
 iris = datasets.load_iris()
 X_train, X_test, y_train, y_test = cross_validation.train_test_split(iris.data, iris.target,