checking in code with random train and test splitting (instead of CV)