cr.sparse.cluster.vq.KMeansSolution¶

class cr.sparse.cluster.vq.KMeansSolution(centroids: jax.Array, assignment: jax.Array, distortion: float, key: jax.Array, iterations: int)[source]¶

The solution for K-means algorithm

Attributes

`assignment`	Current assignment of points to centroids
`centroids`	Current set of centroids
`distortion`	Current mean distance
`iterations`	The number of iterations it took to complete
`key`	The PRNG key seed for the k-means run with least distortion

cr-sparse