Skip to content

Commit d04b42c

Browse files
committed
change init to random
1 parent ef2ef02 commit d04b42c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

model2vec/vocabulary_quantization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def quantize_vocabulary(
3131
# Store the original dtype to restore it later
3232
orig_dtype = embeddings.dtype
3333

34-
kmeans = KMeans(n_clusters=n_clusters, random_state=42, init="k-means++")
34+
kmeans = KMeans(n_clusters=n_clusters, random_state=42, init="random")
3535
cast_embeddings = embeddings.astype(np.float32)
3636
# Fit KMeans to the embeddings
3737
kmeans.fit(cast_embeddings)

0 commit comments

Comments
 (0)