Experiment Tracking Integrations

Comet (air.integrations.comet)

CometLoggerCallback([online, tags, ...])

CometLoggerCallback for logging Tune results to Comet.

See also

See here for an example.

MLflow (air.integrations.mlflow)

MLflowLoggerCallback([tracking_uri, ...])	MLflow Logger to automatically log Tune results and config to MLflow.
setup_mlflow([config, tracking_uri, ...])	Set up a MLflow session.

See also

See here for an example.

Weights and Biases (air.integrations.wandb)

WandbLoggerCallback([project, group, ...])	Weights and biases (https://www.wandb.ai/) is a tool for experiment tracking, model optimization, and dataset versioning.
setup_wandb([config, api_key, api_key_file, ...])	Set up a Weights & Biases session.

See also

See here for an example.