如何在调用 model.predict 后阻止 Keras 打印

Question

我正在训练 keras 序列模型，现在希望预测一个值。我运行以下单行

agent.model.predict(np.array([0,0,0,0]).reshape(1,4))

并在我的笔记本中显示以下输出

1/1 [==============================] - 0s 29ms/step
array([[0.00760011, 0.01811639]], dtype=float32)

如何阻止 Keras 显示输出中的第一行？

Answer 1

正如Gerry P所提到的，为了防止 Keras 打印 model.predict() 的输出，请将 verbose 参数设置为 0，如下所示：

agent.model.predict(np.array([0,0,0,0]).reshape(1,4),verbose = 0)