[23星]fastkmeans:一个超快的PyTorch k-means聚类库,让聚类任务变得轻松高效。亮点:1. 支持CPU和GPU,速度媲美甚至超越FAISS;2. 仅依赖PyTorch和NumP
- 爱可可-爱生活
- 2025-04-21 01:01:27
【[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创造营
'fastkmeans: A fast and efficient k-means implementation for PyTorch, with support for GPU and CPU.'
GitHub: github.com/AnswerDotAI/fastkmeans
聚类算法 PyTorch 高性能计算 AI创造营