setting up the data

getting a training a test set, and a k nearest neighbor model for
comparison