check ch06 on latest softw.

josephine-ytlin · Dec 5, 2019 · 76495d1 · 76495d1
1 parent 7bf4d1f
commit 76495d1
Showing 2 changed files with 66 additions and 57 deletions.
diff --git a/ch06/ch06.ipynb b/ch06/ch06.ipynb
diff --git a/ch06/ch06.py b/ch06/ch06.py
@@ -172,6 +172,7 @@
 
 
 kfold = StratifiedKFold(n_splits=10,
+                        shuffle=True,
                         random_state=1).split(X_train, y_train)
 
 scores = []
@@ -329,6 +330,7 @@
 gs = GridSearchCV(estimator=pipe_svc, 
                   param_grid=param_grid, 
                   scoring='accuracy', 
+                  refit=True,
                   cv=10,
                   n_jobs=-1)
 gs = gs.fit(X_train, y_train)
@@ -339,7 +341,11 @@
 
 
 clf = gs.best_estimator_
-clf.fit(X_train, y_train)
+
+# clf.fit(X_train, y_train) 
+# note that we do not need to refit the classifier
+# because this is done automatically via refit=True.
+
 print('Test accuracy: %.3f' % clf.score(X_test, y_test))
 
 
@@ -499,6 +505,7 @@
 
 
 cv = list(StratifiedKFold(n_splits=3, 
+                          shuffle=True,
                           random_state=1).split(X_train, y_train))
 
 fig = plt.figure(figsize=(7, 5))