/kmeans_pytorch

pytorch implementation of basic kmeans algorithm(lloyd method with forgy initialization) with gpu support

Primary LanguagePython

Stargazers