Keras模型的predict和predict_on_batch方法之间有什么区别?

发布于 2021-01-29 16:54:40

根据keras文档

predict_on_batch(self, x)
Returns predictions for a single batch of samples.

但是,predict在批处理上调用时,标准方法似乎没有任何区别,无论它是一个元素还是多个元素。

model.predict_on_batch(np.zeros((n, d_in)))

是相同的

model.predict(np.zeros((n, d_in)))

numpy.ndarray形状的(n, d_out

关注者
0
被浏览
170
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    区别在于,当您传递的x数据大于一批时。

    predict
    逐批处理 所有数据,预测标签。因此,它在内部进行分批处理,一次进给一批。

    predict_on_batch另一方面,假设您传入的数据恰好是一批,因此将其馈送到网络。它不会尝试拆分它(如果阵列很大,则取决于您的设置,这可能会给您的GPU内存带来问题)



知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看