Guest User

Untitled

a guest
May 22nd, 2018
110
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.95 KB | None | 0 0
  1. !pip install python-telegram-bot
  2.  
  3. import telegram
  4.  
  5. CHAT_ID = 'XXXXXXX'
  6. bot = telegram.Bot('XXXXXX')
  7.  
  8. class MyEarlyStopping(EarlyStopping):
  9. def on_epoch_end(self, epoch, logs=None):
  10. super(MyEarlyStopping, self).on_epoch_end(epoch, logs)
  11. if self.model.stop_training:
  12. try:
  13. bot.send_message(CHAT_ID, 'EarlyStopping')
  14. except Exception as e:
  15. print(e)
  16.  
  17. class MyModelCheckpoint(ModelCheckpoint):
  18. def on_epoch_end(self, epoch, logs=None):
  19. self.prev_best = self.best
  20. super(MyModelCheckpoint, self).on_epoch_end(epoch, logs)
  21. current = logs.get(self.monitor)
  22. try:
  23. if self.monitor_op(current, self.prev_best):
  24. bot.send_message(CHAT_ID, 'Epoch %05d: %s improved from %0.5f to %0.5f' % (epoch + 1, self.monitor, self.prev_best, current))
  25. else:
  26. bot.send_message(CHAT_ID, 'Epoch %05d: %s did not improve from %0.5f' % (epoch + 1, self.monitor, self.prev_best))
  27. except Exception as e:
  28. print(e)
Add Comment
Please, Sign In to add comment