Fang Zhao

这里用的sklearn里的最近邻算法，速度是会比较慢。可以考虑使用基于GPU的算法库来加速，例如[faiss](https://github.com/facebookresearch/faiss)。