改为自己实现的kmeans，很慢