optimization - pytorch中序数多分类的损失函数

标签 optimization neural-network pytorch loss-function

我是 DNN 和 pytorch 的初学者。
我正在处理一个多分类问题,其中我的标签被编码为一个单热向量,比如维度 D .
为此,我使用了 CrossEntropyLoss。但是现在我想修改或更改此类标准,以惩罚与实际值相距甚远的值,例如分类 4 而不是 5 优于分类 2 而不是 5。

Pytorch 中是否已经内置了一个函数来实现这种行为?否则我如何修改 CrossEntropyLoss 来实现它?

最佳答案

这可以帮助你。这是一个 PyTorch 实现序数回归:
https://www.ethanrosenthal.com/2018/12/06/spacecutter-ordinal-regression/

关于optimization - pytorch中序数多分类的损失函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52761533/

相关文章:

amazon-web-services - 使用Amazon EBS节省成本

mysql - 慢速依赖子查询的问题

optimization - 在 Linux 内核中使用可能/不太可能作为返回参数

machine-learning - 他们如何计算 Caffe 中这个卷积网络示例的输出量?

machine-learning - 基于序列数据的流失率预测

c - clang 中的静态预计算优化

machine-learning - Caffe 只训练一个标签

python - 不推荐使用与输入大小 (torch.Size([16, 2])) 不同的目标大小 (torch.Size([16]))

python - 对 pytorch RNN 使用 OHE 查找表的正确方法是什么?

python - 如何进行自定义 torchvision 转换?