Develop a Custom Experiment Tracker

How to develop a custom experiment tracker

This is an older version of the ZenML documentation. To read and view the latest version please visit this up-to-date URL.

Base abstraction in progress!

We are actively working on the base abstraction for the Experiment Tracker, which will be available soon. As a result, their extension is not recommended at the moment. When you are selecting an Experiment Tracker for your stack, you can use one of the existing flavors.

If you need to implement your own Experiment Tracker flavor, you can still do so, but keep in mind that you may have to refactor it when the base abstraction is released.

Build your own custom experiment tracker

If you want to implement your own custom Experiment Tracker, you can follow the following steps:

  1. Create a class which inherits from the BaseExperimentTracker class.

  2. Define the FLAVOR class variable.

Once you are done with the implementation, you can register it through the CLI as:

zenml experiment-tracker flavor register <THE-SOURCE-PATH-OF-YOUR-EXPERIMENT-TRACKER>

ZenML includes a range of Experiment Tracker implementations provided by specific integration modules. You can use them as examples of how you can extend the base Experiment Tracker class to implement your own custom Experiment Tracker:

Last updated