From 922a98097c4b804363cf379c6e38c8c4e4152495 Mon Sep 17 00:00:00 2001
From: Dmitry Kostyaev <d.kostyaev@babo.com>
Date: Thu, 7 Sep 2017 14:45:31 +0300
Subject: [PATCH] update call for sklearn

---
 categorize.py    | 2 +-
 requirements.txt | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/categorize.py b/categorize.py
index 3a8fa3b..5e2f2ac 100644
--- 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
index 18a9c00..bce44c6 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,3 @@
 keras
 tqdm
-sklearn
+scikit-learn>=0.19.0