gensimでtfidfする際にid表示を戻すことはできるのか

Question

gensimを用いてtfidfをしています

wordは形態素解析した単語のリストです

gensim_dictionary = corpora.Dictionary(word)
corpus = [gensim_dictionary.doc2bow(sentence) for sentence in word]
tfidf_model = models.TfidfModel(corpus)
corpus_tfidf = tfidf_model[corpus]
for i in corpus_tfidf:
    print(i)

このようにするともちろん実行結果がprintされてくるのですが
tfidfした結果が
(id, tfidfの値)
のようになります

idではなく、もともとの入力した単語のままにできないのですか？
それか、idを単語にもどすコマンドのようなものがあれば
教えていただけるとありがたいです。

Answer 1

gensim_dictionary[token_id]

で取り出せるはずです。

一覧で見るには以下のように書けばよいでしょう。

for i in corpus_tfidf:
    for j in i:
        print(gensim_dictionary[j[0]], j[1])

Comments

[kohekoh]

ありがとうございます

[kohekoh]

ちなみに、そのままもどすコマンドみたいなのはないですかね？
普通に自分でコード書いて、戻すしかないですか？

[MasashiKimura]

どのような動作を期待されていますか？普通に for文でprint(i)している部分を書き換えれば良いだけではないですか？

[kohekoh]

このコードだとprintしているだけなのですが
実際はこの後に、リストに格納して、値の大きい順に20個にしぼったリストを
新たに作っています
そのリストが、idだとそのあとの処理ができないので、単語に戻したいということです
そこで簡単にもどすコマンドのようなものがあればと思いました

[MasashiKimura]

追記した回答では不足でしょうか？

[kohekoh]

やってみましたが、エラーがでます
ValueError: too many values to unpack (expected 2)

[MasashiKimura]

修正しました。

[kohekoh]

for文で、同じ値だったら
値のところに単語を入れ替える処理をするしかないということですかね

[MasashiKimura]

別に対応する辞書を用意すればよいのではないでしょうか？

[kohekoh]

idを用いない辞書ということですか？

[MasashiKimura]

それはあなたがやりたいこと次第だと思いますよ。