Question

我做了一个在类之间获得最大概率的函数，但是我在问是否有内置函数可以做到这一点，恐怕我的函数是否包含错误。

y_pred_formed=[]
y_pred=model.predict(features)
print(y_pred.shape)
for i in range(y_pred.shape[0]):
    maximum = np.argmax(y_pred[i])
    y_pred_formed.append(maximum)

谢谢

Answer 1

argmax使用一个axis参数，因此您可以将以上内容简化为：

y_pred = np.array([[0.3, 0.4, 0.3], [0.8, 0.2, 0.], [0.2, 0.2, 0.6]]) #example array

print(y_pred)
array([[0.3, 0.4, 0.3],
       [0.8, 0.2, 0. ],
       [0.2, 0.2, 0.6]])

y_pred.argmax(axis=1)
# array([1, 0, 2], dtype=int64)

从概率输出数组中获取预测的类

1 个答案: