Using K-nearest_neighbors_algorithm to classify data