Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- !pip install python-telegram-bot
- import telegram
- CHAT_ID = 'XXXXXXX'
- bot = telegram.Bot('XXXXXX')
- class MyEarlyStopping(EarlyStopping):
- def on_epoch_end(self, epoch, logs=None):
- super(MyEarlyStopping, self).on_epoch_end(epoch, logs)
- if self.model.stop_training:
- try:
- bot.send_message(CHAT_ID, 'EarlyStopping')
- except Exception as e:
- print(e)
- class MyModelCheckpoint(ModelCheckpoint):
- def on_epoch_end(self, epoch, logs=None):
- self.prev_best = self.best
- super(MyModelCheckpoint, self).on_epoch_end(epoch, logs)
- current = logs.get(self.monitor)
- try:
- if self.monitor_op(current, self.prev_best):
- bot.send_message(CHAT_ID, 'Epoch %05d: %s improved from %0.5f to %0.5f' % (epoch + 1, self.monitor, self.prev_best, current))
- else:
- bot.send_message(CHAT_ID, 'Epoch %05d: %s did not improve from %0.5f' % (epoch + 1, self.monitor, self.prev_best))
- except Exception as e:
- print(e)
Add Comment
Please, Sign In to add comment