我使用的是Keras 2.2.4版本,与TensorBoard 1.10.0和TensorFlow 1.10.0一起,在Python 3.6版本上运行。但是我在predict方法上无法使用callbacks回调函数。
我有一个保存的模型,首先我加载了模型,然后尝试预测测试数据并在TensorBoard中显示。结果抛出了一个错误,但根据Keras文档,predict函数的形式应该是这样的:
predict(x, batch_size=None, verbose=0, steps=None, callbacks=None)
TypeError: predict() 函数接收到了意外的关键字参数 ‘callbacks’
回答:
Keras的文档是从git主分支构建的,因此通常包含当前任何Keras版本中没有的功能。对于model.predict
的callbacks回调函数就是这种情况。要使用这些功能,您需要从git主分支安装Keras,如下所示:
pip install git+https://github.com/keras-team/keras.git
在安装主分支之前,您应该先删除其他版本的Keras。