【发布时间】:2018-08-25 15:04:54
【问题描述】:
我想用不同的颜色显示带有 t-SNE 的 doc2vec 模型的文档和单词。我在python2.7中编写了以下代码。但我得到了错误。到目前为止,我还没有在 python 中使用过 t-SNE 和绘图。
modelfile='/home/fl/models/Robust/dimention500/my_model.doc2vec'
wordNum=10000
docNum=1000
words_vec = []
words_label = []
docs_vec = []
docs_label = []
all_vec = []
model = gensim.models.Doc2Vec.load(modelfile)
w2v = model.wv.vocab
d2v = model.docvecs
lengthVoc=len(w2v)
ids=[None]*wordNum
for i in range(10000):
ids[i]=random.randint(1,lengthVoc)
for i, word in enumerate(w2v.keys()):
if i in ids:
all_vec.append(model[word])
words_vec.append(model[word])
for j, doc in enumerate(d2v.doctags.keys()):
if j in docids:
all_vec.append(d2v[doc])
docs_vec.append(d2v[doc])
words = TSNE(n_components=2).fit_transform(numpy.array(words_vec))
docs = TSNE(n_components=2).fit_transform(numpy.array(docs_vec))
all_dw = TSNE(n_components=2).fit_transform(numpy.array(all_vec))
g1=words
g2=docs
data = (g1, g2)
colors = ("red", "blue")
groups = ("word", "document")
for data, color, group in zip(data, colors, groups):
x, y = data
plt.scatter(x, y, c=color, edgecolors='none', label=group)
plt.title('Robust04')
plt.legend(loc=2)
plt.show()
我收到了这个错误:
Traceback (most recent call last):
File "test.py", line 91, in <module>
x, y = data
ValueError: too many values to unpack
当我将代码的以下部分更改为plt.scatter(all_dw[:, 0], all_dw[:, 1])时,它显示了图形,但所有单词和文档都有相同的颜色。
for data, color, group in zip(data, colors, groups):
x, y = data
plt.scatter(x, y, c=color, edgecolors='none', label=group)
【问题讨论】:
标签: python-2.7 plot zip valueerror doc2vec