ある単語をWord2Vecで最も類似度の高い単語で置換したい

Question

図1 ダイエットの類義語一覧

前提・実現したいこと

MeCabで文章を分かち書きして，Word2Vecを用いることである単語と最も類似度の高い単語を置換したい．

置換したい単語を難解語として変数selに入れます．その変数selと，Word2Vecの結果であるresultの中で最も類似度の高い単語を置換したいです．

（例）僕はダイエット中だ　→　僕は減量中だ

具体的には，Word2Vecを用いて類似語を抽出した後にその中で最も類似度の高い語を抽出する方法（図1の"減量"を抽出する方法）と単語の置換方法が分からないです．

該当のソースコード

from gensim.models import word2vec
import csv
import MeCab

model = word2vec.Word2Vec.load("./wiki.model")

dictB = {}

m = MeCab.Tagger("-Owakati")
print("分かち書きしたい文章")
x = input()
print("入力文：" ,x)
word = m.parse(x).split()

print("結果：" ,m.parse(x))

with open('test.csv', mode="w") as f:
    writer = csv.writer(f)
    writer.writerow(word)

with open('test2.csv') as fileB:
    reader = csv.reader(fileB,delimiter=',')
    for row in reader:
        for sel in row:
            dictB[sel] = True

with open('test.csv') as fileA:
    with open('word.csv', mode="w") as r:    //testとtest2を比べて，
        reader = csv.reader(fileA,delimiter=',') //testにしかない単語（難解語）をword.csvに出力
        writer = csv.writer(r)
        for row in reader:
            for sel in row:
                if sel not in dictB:
                    print("難解語：" ,sel)
                    writer.writerow([sel])
                    results = model.wv.most_similar(positive=[sel])
                    for result in results:
                        print(result)

補足情報（FW/ツールのバージョンなど）

Python3.7.1

Comments

[can110]

現状で出来ているコードと具体的に何が分からない（MeCabで分かち書き方／Word2Vecの使い方／文字列の置換方法など）のかを追記ください

[退会済みユーザー]

すみません，コードを載せ直しました．

[can110]

コードから推測すると、置き換えた単語「減量」は標準出力に出すということでよいでしょうか？

[退会済みユーザー]

その通りです．情報不足で申し訳ございません．

Answer 1

提示ソース中においてresults[0]が最も類似度の高い単語です。
あとはこれと元の単語を置き換えればよいです。
置き換えに相当するソース部分が存在しない？（不明）なので、あとはご自身で実装ください。
たとえば以下のようになるかと思います。

# 略
results = model.wv.most_similar(positive=[sel])
print(results[0]) # 最も類似度の高い単語を出力

Comments

[退会済みユーザー]

すみません，コード中に変数resultが重複していたので再び修正しました．
print(results[0])としたら，最も類似度が高い単語が出力されるという認識で合っていますか？

[can110]

most_similarという関数名からそれは明らかであると私は認識しています。
詳細は公式マニュアルで確認ください。

[退会済みユーザー]

承知しました．確認します．

[退会済みユーザー]

回答ありがとうございます．

[can110]

えと、そもそもご自身が「most_similar」関数をコードに書かれているのに
そのコードの意味を他人に尋ねられているので戸惑っています…

[退会済みユーザー]

すみません，https://qiita.com/kenta1984/items/93b64768494f971edf86　を参考に書いたのできちんと意味を理解するまでに至っていませんでいた...

[can110]

現状は「most_similar」関数は理解されたということでよろしいでしょうか？

[退会済みユーザー]

少しは理解したつもりですが，完全に理解したとはいえない状況ですね...

[can110]

あとは実際にいろいろと動かして慣れるしかないと思います。
動かしていくうちに理解できる瞬間が来ますので。

[退会済みユーザー]

わかりました，ありがとうございます．
最後にもう一つだけ良いでしょうか？
ベクトルも表示させず，類義語だけ表示させる方法はあるのでしょうか？

[can110]

こちらでは動作未検証ですのでたぶん「result[0][0]」でよいと思いますが
だめなら「results[0]」のprint結果を提示ください。

[退会済みユーザー]

うまく出力されました．
ありがとうございます．助かりました．