早停法(Early_Stopping)

早停法(Early_Stopping)

目的:在模型达到最好效果的时候停止训练
设定一个监视值(monitor)
设定监视值不再发生改善前允许训练的最大次数

代码实现:
在model.fit之前添加:

early_stop=keras.callbacks.EarlyStopping(monitor='val_loss',patience=10)#模型训练中监视的值为val_loss,当模型在持续训练10个回合后val_loss仍无改善后即终止训练

在model.fit中添加:

history=model.fit(data_2,label,epochs=50,validation_split=0.3,callbacks=[early_stop])