请问一下bert预测时出现keyerror,调试时提示这句有问题 label_id = label_map[example.label]怎么解决?
我也遇到了这个问题,原因是因为使用pandas导出数据为tsv格式时会将列名也保存下来,解决方法是在tsv文件中将列名那一行删掉即可
把label_map和example.label变量print出来看看
看看是不是标签写在前 内容写在后面?
建议检查一下你的tsv文件是否准确,它的第一列是标签而不是索引,默认它的标签只有0和1,第一列有其他数据就会报这个错
我这里也出现了问题,主要是在_get_labels()这个类函数这里return应该改成你的数据集的标签而不是[0,1]