SoftMax 函数用在多分类问题里,把最后的输出结果归一化到概率值。在逻辑回归二分类问题里我们用sigmod函数将结果映射到(0,1)的概率里。在多分类里就可以用softmax函数把输出归一化到概率(0,1)里面。

SoftMax函数的公式:
如果一个多分类模型的输出为 a1, a2, a3...那么Softmax函数的形式就为:对每个样本,它属于类别ii的概率为: