Remove test data from cross validation

[alyciakb]

• Based on feedback, realized our test data was influencing our model fit. Updated the code to use cross validation for max_depth testing with only the training data