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 |
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.2.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 865c648cc52531538131c64df73324265359bd0217229375808289f203f6359c |
|
MD5 | 8ae70ba47ee7904b2c14eee3d22e0433 |
|
BLAKE2b-256 | 372f98debb7c02db6b08f9ab7eab90f933b77116e4785796cfc574bdf430d3b6 |