玩酷网

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

【[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创造营