Generating churn probabilities using KNN