softmax()函数常用做机器学习问题中的分类问题(区分男女,猫狗…)的输出函数。
softmax函数的输出是0.0-1.0之间的实数,并且函数的输出值的总和为1。
我们可以把softmax函数的输出解释为概率。
softmax()函数要进行指数运算,指数函数值容易非常大,使运算结果出现不确定的情况
def softmax(a): #a为数列
c = np.max(a)
exp_a = np.exp(a-c) #溢出对策
sum_exp_a = np.sum(exp_a)
y = exp_a / sum_exp_a
return y
np.sum(y)
1.0