如何显示 Softmax 最高概率 [关闭]

Question

我正在运行以下code。 我试图在预测旁边显示预测标签的概率，但我很难这样做。

Answer 1

我不熟悉 Keras，希望它能工作。

data = test_dataset.take(1)

points, labels = list(data)[0]
points = points[:8, ...]
labels = labels[:8, ...]

# run test data through model
preds = model.predict(points)
print(preds)
preds = tf.reduce_max(preds, -1) #this line you need
print(preds)
preds = tf.math.argmax(preds, -1)
print(preds)

.....

Answer 2

我没有运行代码，但我相信 preds 变量将具有所有概率。

对于您尝试预测的特定输入，如果您想查看最大预测值，可以使用max(pred)。