gpt4 book ai didi

python-3.x - Kmeans fit_predict 与 word2vec

转载 作者:行者123 更新时间:2023-12-04 00:46:18 25 4
gpt4 key购买 nike

我正在尝试使用 kmeans 对我的词向量进行聚类,如所述 here .我正在使用的代码片段

# Set "k" (num_clusters) to be 1/5th of the vocabulary size, or an
# average of 5 words per cluster
word_vectors = model.syn0
num_clusters = word_vectors.shape[0] / 5

# Initalize a k-means object and use it to extract centroids
kmeans_clustering = KMeans( n_clusters = num_clusters )
idx = kmeans_clustering.fit_predict( word_vectors )

我收到以下错误TypeError: 'float' 对象不能解释为整数

有人可以帮忙吗

最佳答案

发现错误。簇数必须是整数,所以我做了以下操作

num_clusters = int(word_vectors.shape[0]/5)

关于python-3.x - Kmeans fit_predict 与 word2vec,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36525317/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com