仅作学习记录~Keras model.fit()参数详解
示例:
callbacks_list = [EarlyStopping(monitor='val_loss', patience=3)] #用early stopping 来防止过拟合
history = model.fit(train_images, train_labels,
epochs=20,
batch_size=200,
validation_data=(test_images, test_labels),
shuffle=True,
callbacks=callbacks_list) #回调函数列表
2.回调函数
在每个training/epoch/batch结束时,如果我们想执行某些任务,例如模型缓存、输出日志、计算当前的acurracy等等,Keras中的callback就派上用场了。
Callbacks(回调函数)是一组用于在模型训练期间指定阶段被调用的函数。可以通过回调函数查看在模型训练过程中的模型内部信息和统计数据。
可以通过传递一个回调函数的list给model.fit()函数,然后相关的回调函数就可以在指定的阶段被调用了。
虽然我们称之为回调“函数”,但事实上Keras的回调函数是一个类,回调函数只是习惯性称呼
Available callbacks
import keras
# 定义callback类
class MyCallback(keras.callbacks.Callback):
def on_train_begin(self, l