Kmeans fit_predict与word2vec

时间:2016-04-10 03:09:08

标签: python-3.x k-means word2vec

我正在尝试使用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' object不能解释为整数

有人可以帮忙吗

0 个答案:

没有答案