Pytorch机器学习——3 神经网络(六)

2022-04-12  本文已影响0人  辘轳鹿鹿

outline

  1. 神经元与神经网络
  2. 激活函数
  3. 前向算法
  4. 损失函数
  5. 反向传播算法
  6. 数据的准备
  7. PyTorch实例:单层神经网络实现

3.4 损失函数

损失函数通常可分为两类,分别用于回归和分类任务。

3.4.1 回归问题

在回归中常用的损失函数是均方差损失(Mean Squared Error, MSE)
L(W,b)=\frac{1}{N}\sum_{i=1}^N(\hat{y}_{i}-y_{i})^2
对于多输出的回归,MSE的公式为:
L(W,b)=\frac{1}{N}\sum_{i=1}^N\sum_{j=1}^M(\hat{y}_{ij}-y_{ij})^2
MSE在回归中使用很广泛,但是它对异常值很敏感,在特定的任务中有时不需要考虑异常值(比方说股票的选择中需要考虑异常值,但是在买房的时候就不需要过多考虑),这时就需要一个损失函数,更关注中位数而不是平均数。这时可以选择平均绝对误差(Mean Absolute Error, MAE),其公式如下:
L(W,b)=\frac{1}{N}\sum_{i=1}^N\sum_{j=1}^M|\hat{y}_{ij}-y_{ij}|

3.4.2 分类问题

上一篇 下一篇

猜你喜欢

热点阅读