2020年11月15日日曜日

word2vecでの色々な結果の表示方法

#ライブラリの読み込み
from gensim.models import Word2Vec

# モデルの読み込み
model = Word2Vec.load("MODEL.model")

#modelベクトルの行列要素を表示(語彙と次元数)
print(model.wv.vectors.shape)

#model.wv.vectorsのうち、1つ目のベクトルを表示
print(model.wv.vectors[0])

#辞書に登録された語彙の数を示す
print(len(model.wv.index2word))

#辞書に登録された語彙を全て示す
print(model.wv.index2word)

#辞書に登録された語彙(1番目から10番目)を示す
print(model.wv.index2word[0:10])
 
参考リンク:Word2Vecの学習済みmodelを扱う備忘録

0 件のコメント:

コメントを投稿