PyPI Stats

Search

All packages
Top packages

Track packages

torch-kmeans


PyPI page
Home page
Author: Jonas K. Falkner
License: MIT
Summary: PyTorch implementations of KMeans, Soft-KMeans and Constrained-KMeans which can be run on GPU and work on (mini-)batches of data.
Latest version: 0.2.0
Required dependencies: importlib-metadata | numpy | torch
Optional dependencies: pre-commit | pytest | pytest-cov | scikit-learn | setuptools | sphinx-rtd-theme

Downloads last day: 566
Downloads last week: 8,893
Downloads last month: 46,005