word2vecの演算結果のフィルタリング

Question

前提・実現したいこと

gensimを使用して、word2vecで単語同士の演算を行なっているのですが、
演算に使用した単語を、演算結果に出てこないようにしたいです。

発生している問題・エラーメッセージ

単語同士の演算結果に、演算に使った同じ単語が［］を付いた状態で出力される問題。
今回で言うと演算結果の上位２位に［料理］と［食材］が出てくる。

該当のソースコード

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=['料理','食材'],topn=20)
for result in results:
    print(result)

演算結果

('[料理]', 0.8916015625)
('[食材]', 0.8163465261459351)
('調理法', 0.8101226091384888)
('[スープ]', 0.7903997898101807)
('[中華料理]', 0.7805976867675781)
('食べ物', 0.7771726250648499)
('[デザート]', 0.7712100744247437)
('[野菜]', 0.7676301002502441)
('[ラーメン]', 0.7663902044296265)
('具材', 0.7656239867210388)
('[寿司]', 0.7650449872016907)
('[サラダ]', 0.7624115347862244

Comments

[quickquip]

質問が書かれてないように見えます。質問はあとから編集できますから、追記するといいかと思います。

Answer 1

以下のように、結果から除外すればよいでしょう。

from gensim.models import KeyedVectors
model_dir = './entity_vector.model.bin'

model = KeyedVectors.load_word2vec_format(model_dir, binary=True)

pstvs = ['料理','食材']
pstvwithps = [f'[{p}]' for p in pstvs]

results = model.most_similar(positive=pstvs,topn=20+len(pstvs))
for result in results:
    if not result[0] in pstvwithps:
    　　　　print(result)

Comments

[GesNer-]

除外できました！
ありがとうございます????