分类问题:激活函数
在做多分类任务的时候,最后一层用了sigmoid激活,结果发现有Loss的情况下,梯度为0,出现sigmoid导致的梯度消失问题,因此改用以下策略使用激活函数:
二分类:Relu+输出层sigmoid
sigmoid:输入一个实数,输出一个实数
多分类:Relu+softmax激活函数
softmax:输入一个向量,最后输出一个向量
在做多分类任务的时候,最后一层用了sigmoid激活,结果发现有Loss的情况下,梯度为0,出现sigmoid导致的梯度消失问题,因此改用以下策略使用激活函数:
二分类:Relu+输出层sigmoid
sigmoid:输入一个实数,输出一个实数
多分类:Relu+softmax激活函数
softmax:输入一个向量,最后输出一个向量