keras模型如何仅预测一个样本?

时间:2017-01-19 08:03:30

标签: python deep-learning

在我的项目中,我想离线培训,这意味着它将以批量方式处理样品(我设置

batch_size=100

在<{1}}中

,我想实时预测一个样本,所以我使用:

model.fit()

但它显示错误:

model.predict(x_real_time, batch_size=1)

有人能告诉我如何解决这个问题吗?感谢

整个代码:

`ValueError: Cannot feed value of shape (1, 3) for Tensor 'input_11:0', which has shape '(165047, 3)'`

1 个答案:

答案 0 :(得分:1)

问题在于输入层。您不应该传递批量大小。如果要使用可变批量大小进行预测,则应传入input_shape而不使用批量大小,然后您可以传入单个样本。

所以:

x = Input(shape=(3,))