muellerzr / Practical-Deep-Learning-for-Coders-2.0

Notebooks for the "A walk with fastai2" Study Group and Lecture Series

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Lesson 3 - Cross-Validation for loop typo

vivekh2000 opened this issue · comments

The following code in Lesson 3 - Cross-Validation appears in the notebook.

skf = StratifiedKFold(n_splits=10, shuffle=True)
for _, val_idx in kf.split(np.array(train_imgs), train_labels):

However, I believe the for loop iterator should be skf.split(..) instead of kf.split(...) as the n_splits=10 is done for skf in the preceding line.
https://walkwithfastai.com/Cross_Validation