Skip to content

Commit 0b59cf1

Browse files
authored
fix: allow KMeans model init parameter as k-means++ alias (#1790)
1 parent f495c84 commit 0b59cf1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

bigframes/ml/cluster.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,8 @@ def __init__(
5959
warm_start: bool = False,
6060
):
6161
self.n_clusters = n_clusters
62-
self.init = init
62+
# allow the alias to be compatible with sklean
63+
self.init = "kmeans++" if init == "k-means++" else init
6364
self.init_col = init_col
6465
self.distance_type = distance_type
6566
self.max_iter = max_iter

0 commit comments

Comments
 (0)