Whitebox - E2E machine learning monitoring

Whitebox is an open source E2E ML monitoring platform with edge capabilities that plays nicely with kubernetes

---

Documentation: https://squaredev-io.github.io/whitebox

Source Code: https://github.com/squaredev-io/whitebox

Roadmap: https://github.com/squaredev-io/whitebox/milestone/2

---

Whitebox is an open source E2E ML monitoring platform with edge capabilities that plays nicely with kubernetes.

The key features are:

• Classification models metrics
• Regression models metrics
• Data / model drift monitoring
• Alerts

Design guidelines:

• Easy: Very easy to set up and get started with.
• Intuitive: Designed to be intuitive and easy to use.
• Pythonic SDK: Pythonic SDK for building your own monitoring infrastructure.
• Robust: Get production-ready MLOps system.
• Kubernetes: Get production-ready code. With automatic interactive documentation.

Installation

Install whitebox server and all of its dependencies in your k8s cluster using helm

helm repo add squaredev https://chartmuseum.squaredev.io/
helm repo update
helm install whitebox squaredev/whitebox

Note: You can also install the server using docker compose. See the docs for more info.

Install the SDK with pip:

pip install whitebox-sdk

How to use

After you are done installing the server and the SDK, you can start using it.

After you get the API key, all you have to do is create an instance of the Whitebox class adding your host and API key as parameters:

from whitebox import Whitebox

wb = Whitebox(host="127.0.0.1:8000", api_key="some_api_key")

Now you're ready to start using Whitebox! Read the documentation to learn more about the SDK.

Set up locally for development

Install packages:

python -m venv .venv
pip install -r requirements.txt
pre-commit install

Run the server:

ENV=dev uvicorn whitebox.main:app --reload

Tests:

• Run: ENV=test pytest -s
• Watch: ENV=test ptw
• Run test coverage ENV=test coverage run -m pytest
• Look at coverage report: coverage report or coverage html to generate an html. To view it in your browser open the htmlcov/index.html file.

Docs

Documentation is hosted bby GitHub here: https://squaredev-io.github.io/whitebox

mkdocs serve -f docs/mkdocs/mkdocs.yml -a localhost:8001

Contributing

We happily welcome contributions to Whitebox. You can start by opening a new issue!