Doc2Vec -- "tag '23943' not seen in training corpus/invalid" 错误

 

训练Doc2Vec模型之后,想要索取文档向量,却出现意外报错:

1
2
3
4
5
for idx, docvec in enumerate(model.docvecs):
    if idx < 17600:
        x_train.append(docvec)
    else:
        x_test.append(docvec)

  

错误情况:

1
KeyError: "tag '23943' not seen in training corpus/invalid"

 

解决方法:

1
2
3
4
5
① model.docvecs.doctag_syn0 (即将弃用)
 
Or
 
② model.docvecs.vectors_docs

  

posted @   Harp_Yestar  阅读(760)  评论(0编辑  收藏  举报
点击右上角即可分享
微信分享提示