We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ef2ef02 commit d04b42cCopy full SHA for d04b42c
model2vec/vocabulary_quantization.py
@@ -31,7 +31,7 @@ def quantize_vocabulary(
31
# Store the original dtype to restore it later
32
orig_dtype = embeddings.dtype
33
34
- kmeans = KMeans(n_clusters=n_clusters, random_state=42, init="k-means++")
+ kmeans = KMeans(n_clusters=n_clusters, random_state=42, init="random")
35
cast_embeddings = embeddings.astype(np.float32)
36
# Fit KMeans to the embeddings
37
kmeans.fit(cast_embeddings)
0 commit comments