CNN_Nested_NER icon indicating copy to clipboard operation
CNN_Nested_NER copied to clipboard
verified publisher icon yhcc

关于损失函数的问题

Open RenDengran opened this issue 2 years ago • 5 comments

我看代码和论文中的损失函数都是BCE，为什么要基于sigmoid函数和给定的0.5阈值进行预测呢，这个多分类任务不应该是基于softmax吗？

RenDengran avatar Jan 18 '24 10:01 RenDengran

因为有个数据集存在一个span同时属于了两个entity类别，索性就都用了sigmoid

yhcc avatar May 20 '24 02:05 yhcc

哥们，在这个模型基础上你尝试了用softmax做多分类任务吗，因为我想识别的是特定的领域的，里面包含的实体全都是非嵌套实体，没有出现一个span同时属于了两个entity类别。

wuxiaohui0 avatar May 28 '24 01:05 wuxiaohui0

哥们，在这个模型基础上你尝试了用softmax做多分类任务吗，因为我想识别的是特定的领域的，里面包含的实体全都是非嵌套实体，没有出现一个span同时属于了两个entity类别。

哥们你成功了吗？求帮助

houyuchao avatar Jul 23 '24 07:07 houyuchao

我看代码和论文中的损失函数都是BCE，为什么要基于sigmoid函数和给定的0.5阈值进行预测呢，这个多分类任务不应该是基于softmax吗？

您好我想问问这部分的代码在哪里，我在二元交叉熵损失函数里有没有找到啊，没看到sigmoid的部分和阈值0.5

houyuchao avatar Jul 23 '24 08:07 houyuchao

https://github.com/yhcc/CNN_Nested_NER/blob/d1c9fae196e2d7692f3d5afd86694dd933b08d71/model/metrics_utils.py#L22C58-L22C63

yhcc avatar Jul 23 '24 11:07 yhcc