predict_KMeans function

Prediction function for the k-means