[23星]fastkmeans:一个超快的PyTorch k-means聚类库,让聚类任务变得轻松高效。亮点:1. 支持CPU和GPU,速度媲美甚至超越FAISS;2. 仅依赖PyTorch和NumP

  • 爱可可-爱生活
  • 2025-04-20 04:29:09
【[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创造营
[23星]fastkmeans:一个超快的PyTorch k-means聚类库,让聚类任务变得轻松高效。亮点:1. 支持CPU和GPU,速度媲美甚至超越FAISS;2. 仅依赖PyTorch和NumPy,安装无忧;3. 采用双分块逻辑,可处理超大数据集,避免内存溢出