Lizhen0628 / text_classification

使用rnn,lstm,gru,fasttext,textcnn,dpcnn,rnn-att,lstm-att,兼容huggleface/transformers,以及以transforemrs作为词嵌入模型,后面接入cnn、rnn、attention等等做文本分类。以及各个模型的对比
491 stars 76 forks source link

KeyError: '[SEP]' #5

Open myf-algorithm opened 4 years ago

myf-algorithm commented 4 years ago

对HAN模型进行训练时,出现报错信息:File "/home1/mayifan/demo/text_classification/model/model.py", line 385, in sep_index = [i for i, num in enumerate(doc_list) if num == self.word_embedding.stoi['[SEP]']] KeyError: '[SEP]'