terrifyzhao / bert-utils

一行代码使用BERT生成句向量,BERT做文本分类、文本相似度计算
Apache License 2.0
1.65k stars 425 forks source link

出现 KeyERROR'0' #11

Open lphu opened 5 years ago

lphu commented 5 years ago

我进行代码的修改, 训练过程中没有出现问题 但是在验证和预测的时候出现看 KEYERROR'0'这个问题, 定位到代码位置是在 label_id = label_map[example.label] 这行 这是什么错误

terrifyzhao commented 5 years ago

label定义的时候和你数据集的label不一样