argmax(axis=1) 用法返回为一维ndarray

最新推荐文章于 2024-04-20 00:06:37 发布

原创最新推荐文章于 2024-04-20 00:06:37 发布 · 772 阅读

0 ·

CC 4.0 BY-SA版权

python实践专栏收录该内容

193 篇文章

订阅专栏

本文展示了一个使用NumPy将预测概率转化为类别标签的例子。通过.argmax()方法沿着指定轴找到最大值的索引来实现这一转换。

predicted_probs = np.array([[0.,   2. ,   0.,      0.,      0.,   0.],[0.,    0.,    3.,       0.,       0.,       0.        ] ,[1.,    0.,     0.,       0.,     0.,      0.   ]])
predictions = predicted_probs.argmax(axis=1)
print("predictions =",predictions)
print("ok")

返回为一维ndarray

predictions = [1 2 0]
ok