gpt4 book ai didi

python - Keras 模型的 predict 和 predict_on_batch 方法有什么区别?

转载 作者:太空狗 更新时间:2023-10-29 17:16:51 30 4
gpt4 key购买 nike

根据keras documentation :

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)))

(形状为(n,d_out)的numpy.ndarray

最佳答案

不同之处在于当您传递大于一批的 x 数据时。

predict将遍历所有数据,逐批,预测标签。因此,它在内部分批进行拆分并一次喂入一批。

predict_on_batch另一方面,假设您传入的数据恰好是一批,因此将其提供给网络。它不会尝试拆分它(根据您的设置,如果数组非常大,这可能会对您的 GPU 内存造成问题)

关于python - Keras 模型的 predict 和 predict_on_batch 方法有什么区别?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44972565/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com