gpt4 book ai didi

python - 如何在列表数组上使用 np argmax?

转载 作者:行者123 更新时间:2023-12-01 01:01:49 31 4
gpt4 key购买 nike

我有一些数据y_hat,看起来像:

[[0. 1. 0. ... 0. 0. 0.]
[0. 1. 0. ... 0. 0. 0.]
[0. 1. 0. ... 0. 0. 0.]
...
[0. 1. 0. ... 0. 0. 0.]
[0. 1. 0. ... 0. 0. 0.]
[0. 1. 0. ... 0. 0. 0.]]

我想获取每行的argmax,这样我最终会得到一个像这样的向量:

[[3]
[8]
[8]
...
[5]
[1]
[7]]

如果我只是执行np.argmax(y_hat),它会返回1

最佳答案

np.argmax accepts an axis keyword argument 。使用它。

列为 axis=0,行为 axis=1

关于python - 如何在列表数组上使用 np argmax?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55737217/

31 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com