Wednesday 16 January 2019

TensorFlow Keras Callback Functions


keras.callbacks.BaseLogger(stateful_metrics=None)
keras.callbacks.TerminateOnNaN()
keras.callbacks.ProgbarLogger(count_mode='samples', stateful_metrics=None)
keras.callbacks.History()

keras.callbacks.ModelCheckpoint(filepath                      ,
                                monitor           = 'val_loss',
                                verbose           = 0         ,
                                save_best_only    = False     ,
                                save_weights_only = False     ,
                                mode              = 'auto'    ,
                                period            = 1         )

keras.callbacks.EarlyStopping(monitor              = 'val_loss',
                              min_delta            = 0         ,
                              patience             = 0         ,
                              verbose              = 0         ,
                              mode                 = 'auto'    ,
                              baseline             = None      ,
                              restore_best_weights = False     )

keras.callbacks.RemoteMonitor(root         = 'http://localhost:9000',
                              path         = '/publish/epoch/end/'  ,
                              field        = 'data'                 ,
                              headers      = None                   ,
                              send_as_json = False                  )

keras.callbacks.LearningRateScheduler(schedule, verbose=0)

keras.callbacks.TensorBoard(log_dir                = './logs',
                            histogram_freq         = 0       ,
                            batch_size             = 32      ,
                            write_graph            = True    ,
                            write_grads            = False   ,
                            write_images           = False   ,
                            embeddings_freq        = 0       ,
                            embeddings_layer_names = None    ,
                            embeddings_metadata    = None    ,
                            embeddings_data        = None    ,
                            update_freq            = 'epoch' )

keras.callbacks.ReduceLROnPlateau(monitor   = 'val_loss',
                                  factor    = 0.1       ,
                                  patience  = 10        ,
                                  verbose   = 0         ,
                                  mode      = 'auto'    ,
                                  min_delta = 0.0001    ,
                                  cooldown  = 0         ,
                                  min_lr    = 0         )

keras.callbacks.CSVLogger(filename, separator=',', append=False)

keras.callbacks.LambdaCallback(on_epoch_begin = None, on_epoch_end = None,
                               on_batch_begin = None, on_batch_end = None,
                               on_train_begin = None, on_train_end = None)

Reference: keras.io

No comments:

Post a Comment