Consider using this https://github.com/ArthurBrussee/KNN
Consider using this https://github.com/ArthurBrussee/KNN