MLNotify is an open-source tool that notifies you when model training is complete via web, mobile, and email notifications – with just one import line.
pip3 install mlnotify
# STEP 1: Import package import mlnotify # STEP 2: Do your thing. my_model.fit(...) # STEP 3: Open https://mlnotify.aporia.com # Enter training ID, and get notified when training is done!
After training hundreds of models, every data scientist knows that waiting for your training to end is tedious. It takes a long time and requires you to Alt+Tab back and forth every now and then to check up on it.
I thought it might be cool to have a tool that will do the waiting for me and send me a notification once it’s done.
To try MLNotify, we’ve prepared a quick snippet for you to copy-paste:
import mlnotify from sklearn import datasets, svm iris = datasets.load_iris() print("Training model...") clf = svm.SVC(tol=1e-100, max_iter=10 ** 9) clf.fit(X=iris.data, y=iris.target)
This should train a simple model for a few minutes. Once training begins, MLNotify will print your unique tracking URL for it:
You can scan the QR, copy the URL, or go to https://mlnotify.aporia.com and enter the code.
You’ll then see the progress of your training. You can enable web, mobile, or email notifications to know exactly when your training finishes.
When training ends, you’ll be notified automatically!
Hurray! No more switching tabs and manually checking for training 🙂
How MLNotify Works
MLNotify is a Python library that hooks into the fit() method of popular ML libraries, and notifies upon completion of the method execution.
The library features a plugin system to extend its functionality.
Now the important question, what should you do while training?
Now that we’ve cleared a lot of time, what should we do with it? It’s an important issue that’s been pondered and debated by data scientists since the beginning of time.
- Grab a cup of coffee
- Discover new MLOps Tools on MLOps.toys
- Read how to build an ML Platform from scratch or Towards Data Science
Contributing to MLNotify
We have plenty of ideas for MLNotify’s roadmap, but we would love to get your help!
The project is available on GitHub: https://github.com/aporia-ai/mlnotify so please feel free to add new features and capabilities.