We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f495c84 commit 0b59cf1Copy full SHA for 0b59cf1
bigframes/ml/cluster.py
@@ -59,7 +59,8 @@ def __init__(
59
warm_start: bool = False,
60
):
61
self.n_clusters = n_clusters
62
- self.init = init
+ # allow the alias to be compatible with sklean
63
+ self.init = "kmeans++" if init == "k-means++" else init
64
self.init_col = init_col
65
self.distance_type = distance_type
66
self.max_iter = max_iter
0 commit comments