Bert 实践遇坑

2020-04-06 22:41:19 +08:00
 MoyeMio

Bert 做中文分类任务,使用的模型是哈工大讯飞开源的中文预训练模型 roberta-wwm-large,按照网上的教程修改了 run-classifier.py ,然后运行之后报了 KeyError 的错。 已经加载了 110000 条数据的情况下报错,感觉应该是 id 为 110000 多的某条数据有问题,但是是什么问题并不清楚,请看官们分析一下。 我将 csv 格式的数据集的 label 一列打印,是 0-10 的 11 个数字没错,并没有出现 keyerror 后面的中文,但是根据报错的信息应该是我的 label 错了,我不知道是什么情况,难顶。

9389 次点击
所在节点    自然语言处理
1 条回复
clockwise9
2020-04-28 14:49:03 +08:00
莫非是字符串编码问题。。。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/659936

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX