前提・実現したいこと
https://analysis-navi.com/?p=2293
こちらのサイトにあるようにdoc2vecで文章の類似度をを調べているのですが最後の結果を出すことができません
発生している問題・エラーメッセージ
IndexError: list index out of range
該当のソースコード
#コサイン類似度の計算+ランキング化 import numpy as np rank_size = 5 v1 = np.linalg.norm(input_vec) cos_sim = [] for v2 in document_vecs: cos_sim.append( np.dot(input_vec,v2)/(v1*np.linalg.norm(v2)) ) doc_sort = np.argsort(np.array(cos_sim))[::-1] cos_sort = sorted(cos_sim,reverse=True) for i in range(rank_size): print(cos_sort[i]) print(documents[doc_sort[i]])
試したこと
ここに問題に対して試したことを記載してください。
補足情報(FW/ツールのバージョンなど)
google colabで実行しています
回答1件
あなたの回答
tips
プレビュー