Want to get notified on the progress of your TensorFlow model training? Enter, a TensorFlow Keras callback to send notifications on the messaging app of your choice.
Project description
tf-notify
Want to get notified on the progress of your TensorFlow model training?
This package provides a tf.keras callback to send notifications to a messaging app of your choice.
Install
The recommended installation is via pip
:
pip install tf-notify
Supported Apps
The following apps are currently supported. But, do check the project frequently, as many more will soon be supported!
Slack | Telegram | Email (SMTP) |
Usage
import tensorflow as tf
from tf_notify import SlackCallback
# define the tf.keras model to add callbacks to
model = tf.keras.Sequential(name='neural-network')
model.add(tf.keras.layers.Dense(1, input_dim=784))
model.compile(
optimizer=tf.keras.optimizers.RMSprop(learning_rate=0.1),
loss="mean_squared_error",
metrics=["mean_absolute_error"],
)
model.fit(
x_train,
y_train,
batch_size=128,
epochs=2,
verbose=0,
validation_split=0.5,
callbacks=[
SlackCallback(webhook_url='https://url.to/webhook')
], # send a Slack notification when training ends!
)
You should see something like this on your Slack:
How to contribute
If you wish to contribute, this is a great place to start!
License
Distributed under the Apache-2.0 license.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for tf_notify-0.3.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8396ee9f4de0bf3c9bd2a49eefa493de1bc060dba71a3a2e5f838f5a1cbe9da4 |
|
MD5 | f86ef326fc10381158eac51eb1040adf |
|
BLAKE2b-256 | 1d2a0217dbad2bb51745ed11f7001a3e8b6bd1efe12eae5c97e31882d7cebc14 |