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

【问题标题】：How can I stop Keras from printing after calling model.predict调用 model.predict 后如何阻止 Keras 打印
【发布时间】：2022-06-13 18:13:18
【问题描述】：

我正在训练一个 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 在输出中显示第一行？

【问题讨论】：

【解决方案1】：

为防止 Keras 打印 model.predict() 的输出，请将详细参数设置为 0，如下所示：

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

【讨论】：