rewrite python kmeans without scipy

facebookresearch / faiss

A library for efficient similarity search and clustering of dense vectors.

https://faiss.ai

MIT License

31.64k stars 3.65k forks source link

rewrite python kmeans without scipy #3873

Closed mdouze closed 2 months ago

mdouze commented 2 months ago

Summary: The previous version required scipy to do the accumulation, which is replaced here with a nifty piece of numpy accumulation. This removes the need for scipy for non-sparse data.

Differential Revision: D62884307

facebook-github-bot commented 2 months ago

This pull request was exported from Phabricator. Differential Revision: D62884307

facebook-github-bot commented 2 months ago

This pull request was exported from Phabricator. Differential Revision: D62884307

facebook-github-bot commented 2 months ago

This pull request has been merged in facebookresearch/faiss@0d7817e88f8cb758f9fe77cadf181baf3cc8e4a3.