Skip to content

Open in Colab

Tracking models end-to-end#

To track a model across all stages of its lifecycle, you can use Neptune for the following:

  • Logging model and run metadata to a central project
  • Grouping models by their stage or other criteria
  • Comparing models to select the best performing model
  • Monitoring a model once in production

To demonstrate this, we provide an example notebook.

Info

The example uses Optuna hyperparameter-optimization to simulate the training and evaluating of multiple XGBoost models, and Evidently to monitor models in production.

However, you can use any library and framework of your choice.

The script can be used as a template to design an automated end-to-end pipeline that covers the entire lifecycle of a model without needing any manual intervention.

Model metadata grouped by stage in the web app

See in Neptune  See script on GitHub 

To download the notebooks and scripts of this example, visit the Neptune examples repo .