diff --git a/Code/Day 11_K-NN.ipynb b/Code/Day 11_K-NN.ipynb index 8fdbe2d..dac425d 100644 --- a/Code/Day 11_K-NN.ipynb +++ b/Code/Day 11_K-NN.ipynb @@ -145,7 +145,7 @@ "metadata": {}, "outputs": [], "source": [ - "from sklearn.cross_validation import train_test_split\n", + "from sklearn.model_selection import train_test_split\n", "X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.25, random_state = 0)" ] },