Skip to content

init="kmeans++" is not documented as a valid option in KMeans #7352

@aamijar

Description

@aamijar

https://github.com/rapidsai/cuml/blob/branch-25.12/python/cuml/cuml/cluster/kmeans.pyx#L55

I found that there is actually a kmeans++ initialization method in our KMeans implementation which works better than the other options for my use case, but it is not in the doc string.

I also believe this is not handled correctly in cuml.accel since scalable-k-means++ and kmeans++ are treated the same

https://github.com/rapidsai/cuml/blob/branch-25.12/python/cuml/cuml/cluster/kmeans.pyx#L418-L419

Metadata

Metadata

Assignees

Labels

algo: kmeansbugSomething isn't workingcuml-accelIssues related to cuml.accel

Type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions