跳到正文
Hacker News· mb14·· 3 小时前AI 评分26

Show HN:面向 PyTorch 张量的 mini-batch k-means 聚类库

Show HN: Mini-batch k-means clustering for PyTorch tensors

AI 导读

minibatch-kmeans 发布,为 PyTorch 张量提供支持 CPU 和 GPU 的 mini-batch k-means 聚类,算法遵循 Sculley 的 Web-Scale K-Means Clustering(WWW 2010)。

来源:Hacker News · github.com