2020年11月16日月曜日

word2vecから語彙をファイルに保存する。

 語彙の保存(例:最初から1000番目まで)

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

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

#vocabの定義
vocab = model.wv.index2word

#ファイル(OUTPUT.txt)に書き込む
with open('OUTPUT.txt', 'w') as f:
    print(vocab[0:1000], file = f)

 

0 件のコメント:

コメントを投稿