Skip to content

Commit 350ee82

Browse files
committed
update n_init use
1 parent b09611a commit 350ee82

File tree

2 files changed

+666
-403
lines changed

2 files changed

+666
-403
lines changed

cpp/include/cuvs/cluster/kmeans.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,9 @@ struct params : base_params {
103103
raft::random::RngState rng_state{0};
104104

105105
/**
106-
* Number of instance k-means algorithm will be run with different seeds.
106+
* Number of instance k-means algorithm will be run with different seeds. For MiniBatch mode,
107+
* this is the number of different initializations to try, but the algorithm is only run once with
108+
* the best initialization.
107109
*/
108110
int n_init = 1;
109111

0 commit comments

Comments
 (0)