https://ksml4.com/customizing-knn-classifier-with-sklearn/