问题描述
我正在为分类任务训练 BertForSequenceClassification。我的数据集由“包含不利影响”(1)和“不包含不利影响”(0)组成。数据集包含所有 1,然后是 0(数据未打乱)。对于训练,我已经洗牌了我的数据并获得了 logits。据我所知,logits 是 softmax 之前的概率分布。一个示例 logit 是 [-4.673831,4.7095485]。第一个值是否对应于标签 1(包含 AE),因为它首先出现在数据集中,还是标签 0。感谢任何帮助。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)