Receive notifications about your model training on your messaging app!
Project description
Keras Notification Callback
A Keras callback which sends information about your model training, on various messaging platforms.
Installation
Using pip
:
pip install keras-webhook-callback
Usage
Import the required module and add it to the list callbacks while training your model.
Example:
>>> from keras_webhook_callback import TelegramCallback
>>> telegram_callback = TelegramCallback('<BotToken>',
'<ChatID>',
'CNN Model',
['loss', 'val_loss'],
['accuracy', 'val_accuracy'],
True)
>>> model.fit(x_train, y_train,
batch_size=32,
epochs=10,
validation_data=(x_test, y_test),
callbacks=[telegram_callback])
Telegram
- Create a telegram bot using BotFather
- Search for @BotFather on telegram.
- Send
/help
to get list of all commands. - Send
/newbot
to create a new bot and complete the setup. - Copy the bot token after creating the bot.
- Get the chat ID
- Search for the bot you created and send it any random message.
- Go to this URL
https://api.telegram.org/bot<BOT_TOKEN>/getUpdates
(replace <BOT_TOKEN> with your bot token) - Copy the
chat id
of the user you want to send messages to.
- Use the
TelegramCallback()
class.
TelegramCallback(bot_token=None, chat_id=None, modelName='model', loss_metrics=['loss'], acc_metrics=[], getSummary=False):
Arguments:
bot_token
: unique token of Telegram bot{str}
chat_id
: Telegram chat id you want to send message to{str}
modelName
: name of your model{str}
loss_metrics
: loss metrics you want in the loss graph{list of strings}
acc_metrics
: accuracy metrics you want in the accuracy graphs{list of strings}
getSummary
: Do you want message for each epoch (False) or a single message containing information about all epochs (True).{bool}
Slack
- Create a Slack workspace
- Create a new channel
- Search for the Incoming Webhooks in the Apps and install it.
- Copy the Webhook URL
- Use the
SlackCallback()
class.
SlackCallback(bot_token=None, chat_id=None, modelName='model', loss_metrics=['loss'], acc_metrics=[], getSummary=False):
Arguments:
webhookURL
: unique webhook URL of the app{str}
channel
: channel name or username you want to send message to{str}
modelName
: name of your model{str}
loss_metrics
: loss metrics you want in the loss graph{list of strings}
acc_metrics
: accuracy metrics you want in the accuracy graph{list of strings}
getSummary
: Do you want message for each epoch (False) or a single message containing information about all epochs (True).{bool}
Sending images in Slack is not supported currently.
ToDo
- Zulip
- Messages
Motivation
As the Deep Learning models are getting more and more complex and computationally heavy, they take a very long time to train. During my internship, people used to start the model training and left it overnight. They could only check its progress the next day. So I thought it would be great if there was a simple way to get the training info remotely on their devices.
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for keras_webhook_callback-0.2.3.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | dab7ba8a5b5db756b7b2a35d0f5e4bec90d113fe44a1307893d1f9820febf372 |
|
MD5 | e5776eb6d0cc5aff56c90c05790ab75f |
|
BLAKE2b-256 | cc9d7983172005aee831ff8793d81392fb1c4d1ea5d43eee902f96db542b7dcc |