2016-04-10 8 views
2

hereのようにkmeansを使用して単語ベクトルをクラスタリングしようとしています。私は次のようなエラーに はTypeError取得していますword2vecでfit_predictを実行する

# 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) 

使用しています コードスニペット:誰かがエラーを発見

答えて

2

を助けてくださいでした「フロート」オブジェクトは、整数

として解釈することはできません。私は、次の

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

+0

UGGGGHをしたように、クラスタの数は整数でなければなりません!ありがとうございました!私は自分のコードでエラーの同じ原因を見つけていないので、とてもダムを感じます:P –

+0

いいえ。間違いは起こります。私たちが認識し、学び、前進する限り、間違いはありません:D – LGG

+1

wordVectors.shape [0] // 5も使えます。 //演算子は結果をintに強制します – icaro56

関連する問題