x=pred_vec.argmax(axis=-1)
x
o/p-array([51], dtype=int64)
我需要提取51,以便可以用来显示其被标记为的类。
答案 0 :(得分:0)
似乎您正在尝试使用python和numpy(基于标记??)来查找最大值的索引。如果是这样,请尝试以下操作:
In [1]: import numpy as np
In [3]: x = np.random.randint(1, 200, (50))
In [4]: x
Out[4]:
array([186, 117, 150, 68, 118, 197, 169, 48, 90, 67, 117, 90, 133,
141, 175, 194, 21, 107, 103, 61, 154, 87, 42, 59, 67, 66,
198, 76, 190, 7, 188, 146, 158, 42, 106, 157, 1, 114, 185,
58, 169, 89, 120, 81, 28, 49, 197, 73, 169, 142])
In [5]: index_max = np.argmax(x)
In [6]: x[index_max]
Out[6]: 198
看看sorting, searching and counting page以获得更多选项。