数据集报错

[1125690278]

用你的数据集thucnews跑多分类是OK的，用自己的数据集一直出现这个错误，请问数据集需要怎么处理吗？
Traceback (most recent call last):
File "run_fastbert.py", line 652, in
main()
File "run_fastbert.py", line 589, in main
result = evaluate(args, False, False)
File "run_fastbert.py", line 445, in evaluate
p = confusion[i,i].item()/confusion[i,:].sum().item()
ZeroDivisionError: division by zero

[hiber-niu]

我也遇到同样的问题，请问解决了吗？

[1125690278]

我也遇到同样的问题，请问解决了吗？

暂时还没解决，解决了请告知一下

[BshoterJ]

试试调小些bs呢？可能是上来把所有数据训成某一类了

[Franklwl]

line 455-459:
for i in range(confusion.size()[0]):
mu = confusion[i, :].sum().item()
p = confusion[i, i].item() / mu if mu != 0. else 0
r = confusion[i, i].item() / mu if mu != 0. else 0
f1 = 2 * p * r / (p + r) if (p + r) != 0. else 0
is OK!