Python library for KNN algorithm