Q&A
実現したいこと
pythonのgensimを用いて、word2vecを使っているのですが、
類似度を指定して出力させたいです。
（できれば、類似度0.3以上0.7以下でやりたい）
発生している問題
特に無し
該当のソースコード
from gensim.models import KeyedVectors
model_dir = './entity_vector.model.bin'
model = KeyedVectors.load_word2vec_format(model_dir, binary=True)
results = model.most_similar(positive='SONY',topn=20,)
for result in results:
print(result)
試したこと
topn引数で検索結果上位１０００個のリストからifで類似度条件を設定して
もう一つのリストに格納した
補足情報（FW/ツールのバージョンなど）
MAC
gensim 3.8.3
回答1件
0
使ったことがないので確実ではありませんが、たぶん以下のようなコードでできるでしょう。
python
1results = model.most_similar(positive='SONY',topn=1000,) 2results_3_7 = [r for r in results if 0.3 <= r[1] <= 0.7]
投稿2021/10/27 06:28
総合スコア24670
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。