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