在每個training/epoch/batch結(jié)束時,如果我們想執(zhí)行某些任務(wù),例如模型緩存光戈、輸出日志、計算當(dāng)前的auc等等遂赠,Keras中的callback就派上用場了久妆。
Example記錄每個batch的損失函數(shù)值
如上述例子,我們可以繼承keras.callbacks.Callback來定義自己的callback,只需重寫其中的6個方法即可
on_train_begin
on_train_end
on_epoch_begin
on_epoch_end
on_batch_begin
on_batch_end
可在這6個方法中定義自己想要的屬性跷睦,通過self.model可以訪問模型本身筷弦,self.params可以訪問訓(xùn)練參數(shù)。
可能有用的屬性
self.validation_datavalidate數(shù)據(jù)集
self.validation_data[0]為X
self.validation_data[1]為y
轉(zhuǎn)載自:http://www.reibang.com/p/eb4e76e239df