autoliuweijie / K-BERT

Source code of K-BERT (AAAI2020)
https://ojs.aaai.org//index.php/AAAI/article/view/5681
949 stars 212 forks source link

是否可以多分类 #79

Open buthi opened 2 years ago

buthi commented 2 years ago

您好,我正在复现您的代码,请问对文本分类任务是只能实现二分类还是可以实现多分类呢。我用自己的多类数据训练时报错: Traceback (most recent call last): File "run_kbert_cls.py", line 578, in main() File "run_kbert_cls.py", line 557, in main result = evaluate(args, False) File "run_kbert_cls.py", line 393, in evaluate print("Acc. (Correct/Total): {:.4f} ({}/{}) ".format(correct/len(dataset), correct, len(dataset))) ZeroDivisionError: division by zero

buthi commented 2 years ago

已解决 刚开始没仔细看

000163 commented 1 year ago

已解决 刚开始没仔细看

請問您後來能使用多分類任務嗎

buthi commented 1 year ago

已解决 刚开始没仔细看

請問您後來能使用多分類任務嗎

可以

SamIp523 commented 1 year ago

您好,不知道是否可以请教您是修改哪一部分以实现多分类呢?