check ch06 on latest softw.

sakthi5006 · Dec 5, 2019 · be77b92 · be77b92
1 parent 76495d1
commit be77b92
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 266 deletions.
diff --git a/ch06/ch06.ipynb b/ch06/ch06.ipynb
diff --git a/ch06/ch06.py b/ch06/ch06.py
@@ -171,9 +171,7 @@
 
 
 
-kfold = StratifiedKFold(n_splits=10,
-                        shuffle=True,
-                        random_state=1).split(X_train, y_train)
+kfold = StratifiedKFold(n_splits=10).split(X_train, y_train)
 
 scores = []
 for k, (train, test) in enumerate(kfold):
@@ -504,9 +502,7 @@
 X_train2 = X_train[:, [4, 14]]
 
 
-cv = list(StratifiedKFold(n_splits=3, 
-                          shuffle=True,
-                          random_state=1).split(X_train, y_train))
+cv = list(StratifiedKFold(n_splits=3).split(X_train, y_train))
 
 fig = plt.figure(figsize=(7, 5))