KK的博客

原创 Pytorch中的CrossEntropy

Pytorch中的CrossEntropy表达式如何计算得到输出Pytorch中的CrossEntropy表达式(1) 二分类在二分的情况下，模型最后需要预测的结果只有两种情况，对于每个类别我们的预测得到的概率为 p 和1-p 。此时表达式为：其中：yi —— 表示样本i的label，正类为1，负类为0pi —— 表示样本i预测为正的概率(2) 多分类多分类的情况实际上就是...