From 9cb9f6ea83506b38a0ec0609c272e815fe8a7f97 Mon Sep 17 00:00:00 2001
From: Aurelien Geron <aurelien.geron@gmail.com>
Date: Sat, 5 Nov 2016 14:26:29 +0100
Subject: [PATCH] Move to sklearn 0.18

---
 04_training_linear_models.ipynb | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index 23bd704..e2c282a 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.ipynb
@@ -834,7 +834,7 @@
    ],
    "source": [
     "from sklearn.metrics import mean_squared_error\n",
-    "from sklearn.cross_validation import train_test_split\n",
+    "from sklearn.model_selection import train_test_split\n",
     "\n",
     "def plot_learning_curves(model, X, y):\n",
     "    X_train, X_val, y_train, y_val = train_test_split(X, y, test_size=0.2, random_state=10)\n",
@@ -1843,7 +1843,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.5.1"
+   "version": "3.5.2"
   },
   "nav_menu": {},
   "toc": {