Hierarchical KMeans does not output a non-zero inertia value.
import cuvs.cluster.kmeans as cuvs_kmeans
# Set KMeans parameters:
cuvs_kmeans_params = cuvs_kmeans.KMeansParams(n_clusters=10, max_iter=None, hierarchical=True)
# Train cuVS kmeans model:
centroids, inertia, n_iter = cuvs_kmeans.fit(cuvs_kmeans_params, X)
The value for inertia is currently always zero. Inertia can be computed separately using the centroids and labels, but is memory intensive if done all at once.
from cuvs.cluster.kmeans import cluster_cost
kmeans_result['inertia'] = cluster_cost(X, centroids)
Hierarchical KMeans does not output a non-zero
inertiavalue.The value for
inertiais currently always zero. Inertia can be computed separately using the centroids and labels, but is memory intensive if done all at once.