Depp Learning note Day1

1.softmax函数的优化来防止溢出

2.np.argmax()函数的使用

返回值为数组中最大值的索引

 a = [1, 2, 3, 4, 3, 7]
 print(np.argmax(a))
 

若np.argmax()中的参数为矩阵，则需要添加axis参数

e.g.

 a = [[12, 2, 3], [9, 14, 7]]

 # 显示每一行(横向)的最大值的索引
 print(np.argmax(a, axis=1))
 # output: [0 1]

 # 显示每一列(纵向)的最大值的索引
 print(np.argmax(a, axis=0))
 # output: [0 1 1]