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.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | fff45fc137a0481ef9c615aa741ecbe2dfe1345ba320f3934b71a5bb6b077330 |
|
MD5 | b7db00921f82b35f55ee899714dce075 |
|
BLAKE2b-256 | 4236e6f5ec533d6d75707bb7bd054d41aedd2aefccf04ff75a51a25eb3193e7a |