tensorflow官网:https://tensorflow.google.cn/api_docs/python/tf/keras/callbacks?hl=en
部分callbacks函数的实现如下:
logdir = './callbacks'
if not os.path.exists(logdir):
os.mkdir(logdir)
output_model_file = os.path.join(logdir,
'fashion_mnist_model.h5')
callbacks = [
keras.callbacks.TensorBoard(logdir),
keras.callbacks.ModelCheckpoint(output_model_file,save_best_only=True)
keras.callbacks.EarlyStopping(patience=5,min_delta=1e-3)
]
history = model.fit(x_train_scaler,y_train,epochs=5,validation_data=(x_valid_scaler,y_valid),callbacks = callbacks)
模型训练完成之后,打开命令行输入:
~/username/tensorflow_test $ tensorboard --logdir == '文件夹的名字' #callbacks所在的文件夹
返回一个网址,打开网址即可查看。