cr.sparse.cluster.vq.find_new_centroids_jit

cr.sparse.cluster.vq.find_new_centroids_jit(assignment, points, k)

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