search

DeMoriarty / fast_pytorch_kmeans

This is a pytorch implementation of k-means clustering algorithm
MIT License
284 stars 38 forks source link

Fix init methods #19

Closed moetayuko closed 6 months ago

fmmoret commented 6 months ago

@DeMoriarty

sample_size comes in as None from the KMeans. Kmeans class depends on it being None but init methods depend on it being -1 if not specified.

DeMoriarty commented 6 months ago

closing due to #25