我在Python 3,6版本中使用tensorboard 1.10.0和tensorflow 1.10.0的Keras 2.2.4版本。但是我不能在预测方法上使用回调。
我有一个保存的模型,因此首先加载模型,然后尝试预测测试数据并将其显示在张量板上。抛出一个错误,但是根据keras文档,预测函数看起来像:
predict(x,batch_size = None,verbose = 0,steps = None,callbacks = None)
import keras.callbacks
from keras.models import load_model
model = load_model(strPath_model)
tb_test = keras.callbacks.TensorBoard(log_dir=strPath_model_test_logs,histogram_freq=0, write_graph=True, write_images=True)
y_test = model.predict(test_val_X, verbose=1, callbacks=[tb_test])
TypeError:predict()收到了意外的关键字参数“回调”
答案 0 :(得分:1)
keras文档是从git master构建的,因此它通常包含不属于任何当前Keras版本的功能。 model.predict
的回调就是这种情况,要使用它们,您必须从git master安装Keras:
pip install git+https://github.com/keras-team/keras.git
在安装master之前,您应该删除其他Keras的版本。