cr.sparse.cluster.vq.find_new_centroids

cr.sparse.cluster.vq.find_new_centroids(assignment, points, k)[source]

Finds new centroids based on current assignment

Parameters

  • assignment (jax.numpy.ndarray) – current assignment of each point to a specific cluster

  • points (jax.numpy.ndarray) – Each row of the points matrix is a point.

  • k (int) – The number of clusters