本文展示了一个使用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
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
