bert预测时出现keyerror

请问一下bert预测时出现keyerror,调试时提示这句有问题  label_id = label_map[example.label]怎么解决?

我也遇到了这个问题,原因是因为使用pandas导出数据为tsv格式时会将列名也保存下来,解决方法是在tsv文件中将列名那一行删掉即可

把label_map和example.label变量print出来看看

看看是不是标签写在前 内容写在后面?

建议检查一下你的tsv文件是否准确,它的第一列是标签而不是索引,默认它的标签只有0和1,第一列有其他数据就会报这个错

我这里也出现了问题,主要是在_get_labels()这个类函数这里return应该改成你的数据集的标签而不是[0,1]