Skip to content

Commit

Permalink
update n_init for kmeans (#153)
Browse files Browse the repository at this point in the history
* update kmeans

* read cuml params

* reset normalize

---------

Co-authored-by: md.shafiul.alam <[email protected]>
  • Loading branch information
md-shafiul-alam and md.shafiul.alam authored Aug 28, 2024
1 parent bbe99dd commit 9b94b00
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion configs/regular/kmeans.json
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
"estimator": "KMeans",
"estimator_params": {
"n_clusters": "[SPECIAL_VALUE]auto",
"n_init": 10,
"n_init": 1,
"max_iter": 30,
"tol": 1e-3,
"random_state": 42
Expand Down

0 comments on commit 9b94b00

Please sign in to comment.