【发布时间】:2022-12-11 02:42:54
【问题描述】:
我有一个关于 pytorch 优化器的初学者高级问题,即是否有一种非自定义的方法来优化真阳性? 假设我有一个标签列表:
labels=[0,0,0,0,0,0,0,0,0,1]
我想要一个模型以一种有利于真阳性的方式适合这些标签——也就是我真的希望它返回“1”作为最后一项,我并不那么担心其他项目的误报“1”在列表中。
我想我可以通过定义一个自定义损失函数来解决这个问题,该损失函数会给我真阳性/误报的权重,但我想知道是否有开箱即用的解决方案。
亲切的问候。
【问题讨论】:
标签: python pytorch loss-function