python人工智能编程笔记总结

np.argmax(prob_scores, axis=1)

并不是取每一列的最大值。

而是找出每行的最大值 ，返回最大值的列索引

prob_scores=\[ 0.97875 0.021254

0.99951 0.00049028

0.98667 0.01333\]

pred_index = np.argmax(prob_scores, axis=1)

得到:0,0,0

pred_score = np.max(prob_scores, axis=1)

取每行的最大值