【[23星]fastkmeans:一个超快的PyTorchk-means聚类库

爱生活爱珂珂 2025-04-18 16:01:40

【[23星]fastkmeans:一个超快的PyTorch k-means聚类库,让聚类任务变得轻松高效。亮点:1. 支持CPU和GPU,速度媲美甚至超越FAISS;2. 仅依赖PyTorch和NumPy,安装无忧;3. 采用双分块逻辑,可处理超大数据集,避免内存溢出】

'fastkmeans: A fast and efficient k-means implementation for PyTorch, with support for GPU and CPU.'

GitHub: github.com/AnswerDotAI/fastkmeans

聚类算法 PyTorch 高性能计算 AI创造营

0 阅读:0
爱生活爱珂珂

爱生活爱珂珂

感谢大家的关注