update call for sklearn

kostyaev · Sep 7, 2017 · 922a980 · 922a980
1 parent af04bad
commit 922a980
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/categorize.py b/categorize.py
@@ -4,7 +4,7 @@
 from keras.preprocessing import image
 import tqdm
 import glob
-from sklearn.cross_validation import train_test_split
+from sklearn.model_selection import train_test_split
 from sklearn.svm import LinearSVC
 from sklearn.metrics import accuracy_score
 import os

diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,3 @@
 keras
 tqdm
-sklearn
+scikit-learn>=0.19.0