diff --git a/notebooks/05.07-Support-Vector-Machines.ipynb b/notebooks/05.07-Support-Vector-Machines.ipynb index 31cf9508b..d8011093c 100644 --- a/notebooks/05.07-Support-Vector-Machines.ipynb +++ b/notebooks/05.07-Support-Vector-Machines.ipynb @@ -778,10 +778,10 @@ "outputs": [], "source": [ "from sklearn.svm import SVC\n", - "from sklearn.decomposition import RandomizedPCA\n", + "from sklearn.decomposition import PCA\n", "from sklearn.pipeline import make_pipeline\n", "\n", - "pca = RandomizedPCA(n_components=150, whiten=True, random_state=42)\n", + "pca = PCA(n_components=150, whiten=True, svd_solver='randomized', random_state=42)\n", "svc = SVC(kernel='rbf', class_weight='balanced')\n", "model = make_pipeline(pca, svc)" ] @@ -801,7 +801,7 @@ }, "outputs": [], "source": [ - "from sklearn.cross_validation import train_test_split\n", + "from sklearn.model_selection import train_test_split\n", "Xtrain, Xtest, ytrain, ytest = train_test_split(faces.data, faces.target,\n", " random_state=42)" ] @@ -832,7 +832,7 @@ } ], "source": [ - "from sklearn.grid_search import GridSearchCV\n", + "from sklearn.model_selection import GridSearchCV\n", "param_grid = {'svc__C': [1, 5, 10, 50],\n", " 'svc__gamma': [0.0001, 0.0005, 0.001, 0.005]}\n", "grid = GridSearchCV(model, param_grid)\n",