From 3bde0353885cc61bb54d7244a1b9feb3404cbbd0 Mon Sep 17 00:00:00 2001 From: Greg Baker Date: Sun, 25 Jun 2017 12:08:09 -0700 Subject: [PATCH] fix scikit-learn deprecation warnings in SVC unit --- notebooks/05.07-Support-Vector-Machines.ipynb | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/notebooks/05.07-Support-Vector-Machines.ipynb b/notebooks/05.07-Support-Vector-Machines.ipynb index d3ae901a9..fe51f9f37 100644 --- a/notebooks/05.07-Support-Vector-Machines.ipynb +++ b/notebooks/05.07-Support-Vector-Machines.ipynb @@ -775,10 +775,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)" ] @@ -798,7 +798,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)" ] @@ -829,7 +829,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",