损失函数(loss)

5-损失函数(Loss) 可能影响网络训练失败_哔哩哔哩_bilibili

回归问题是模型给出一个预测值

分类问题,我们一般使用one-hot进行class的分类

输入特征 x 进入神经网络,经过加权求和、偏置、激活函数,最后输出三个类别的预测值。真实类别用 one-hot 向量表示,模型输出也应该是一个长度为 3 的向量,然后通过比较预测向量和真实 one-hot 标签来训练模型。

softmax 0-1

第一个是回归问题的loss函数,预测结果大多是连续的

第二个是交叉熵的loss函数,而分类预测大多是离散,比如0,1、

在pytorch中,soft-max被内置在了Cross-entropy

总结:可以通过改变损失函数,改变optimization的优化难度