我在运行以下代码。我试图在预测结果旁边显示预测标签的概率，但我发现这很难做到。

回答：

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

data = test_dataset.take(1)points, labels = list(data)[0]points = points[:8, ...]labels = labels[:8, ...]# 运行测试数据通过模型preds = model.predict(points)print(preds)preds = tf.reduce_max(preds, -1) #你需要这行print(preds)preds = tf.math.argmax(preds, -1)print(preds).....