pythonプログラムのエラー対処方をご教授いただきたく存じます。
以下のプログラミングは、ある文章から単語の意味ベクトルが学習されたモデル(sys.argv[1])を、scikit-learnライブラリに実装されているt-SNEを用いて次元圧縮してから、平面上にプロットし図(sys.argv[2])として保存するプログラミングです。
python
1# -*- coding: utf-8 -*- 2 3import sys 4import matplotlib.pyplot as plt 5import numpy as np 6from gensim.models import Word2Vec 7from sklearn.manifold import TSNE 8 9# 学習したモデルのロード 10model = Word2Vec.load(sys.argv[1]) 11 12# 「砂糖」に近い単語を50個取得して視覚化を行う 13words = [word for (word, score) in model.wv.most_similar('砂糖', topn=50)] 14words.append('砂糖') 15vectors = np.vstack([model[word] for word in words]) 16 17# t-SNEで意味ベクトルを2次元の空間にマップする 18tsne = TSNE(n_components=2) 19Y = tsne.fit_transform(vectors) 20 21# 散布図上にベクトルをあらわす点を描画する 22x_coords = Y[:, 0] 23y_coords = Y[:, 1] 24plt.scatter(x_coords, y_coords) 25 26# 散布図上に単語を描画する 27for (word, x, y) in zip(words, x_coords, y_coords): 28 plt.annotate(word, xy=(x, y), xytext=(5, -10), textcoords='offset points') 29 30# 散布図を書き出す 31plt.savefig(sys.argv[2])
実行した結果、以下のようなエラーが発生します。
Traceback (most recent call last): File "visualize.py", line 15, in <module> vectors = np.vstack([model[word] for word in words]) File "visualize.py", line 15, in <listcomp> vectors = np.vstack([model[word] for word in words]) TypeError: 'Word2Vec' object is not subscriptable
調べたところ、「'〇〇' object is not subscriptable」は、「〇〇オブジェクトの添字不可」という意味らしく、リストの表記に何か誤った表記があるかと思われます。
しかし、調べても修正方法がわかりませんでした。
お忙しい中恐縮ですが、どなたかお力添え頂ければ幸いです。
回答1件
あなたの回答
tips
プレビュー