Testing and training data in R (STAT 320 week 8, video 6 of 8)