This feature is in Public Preview in the following AWS regions:
- US East (Ohio)
- EU (Frankfurt)
- EU (Ireland)
- Canada (Central)
- Asia Pacific (Mumbai)
- Asia Pacific (Sydney)
- Asia Pacific (Tokyo)
Databricks Autologging is a no-code solution that extends MLflow automatic logging to deliver automatic experiment tracking for machine learning training sessions on Databricks. With Databricks Autologging, model parameters, metrics, files, and lineage information are automatically captured when you train models from a variety of popular machine learning libraries. Training sessions are recorded as MLflow tracking runs. Model files are also tracked so you can easily log them to the MLflow Model Registry and deploy them for real-time scoring with MLflow Model Serving.
The following video shows Databricks Autologging with a scikit-learn model training session in an interactive Python notebook. Tracking information is automatically captured and displayed in the Experiment Runs sidebar and in the MLflow UI.
When you attach an interactive Python notebook to a Databricks cluster, Databricks Autologging calls mlflow.autolog() to set up tracking for your model training sessions. When you train models in the notebook, model training information is automatically tracked with MLflow Tracking. For information about how this model training information is secured and managed, see Security and data management.
The default configuration for the mlflow.autolog() call is:
mlflow.autolog( log_input_examples=False, log_model_signatures=True, log_models=True, disable=False, exclusive=True, disable_for_unsupported_versions=True, silent=True )
To use Databricks Autologging, train a machine learning model in a supported framework using an interactive Databricks Python notebook. Databricks Autologging automatically records model lineage information, parameters, and metrics to MLflow Tracking. You can also customize the behavior of Databricks Autologging.
Databricks Autologging is not applied to runs created using the
MLflow fluent API with
mlflow.start_run(). In these cases, you must call
mlflow.autolog() to save autologged content
to the MLflow run. See Track additional content.
To customize logging, use mlflow.autolog().
This function provides configuration parameters to enable model logging (
collect input examples (
log_input_examples), configure warnings (
silent), and more.
To track additional metrics, parameters, files, and metadata with MLflow runs created by Databricks Autologging, follow these steps in a Databricks interactive Python notebook:
- Call mlflow.autolog()
- Start an MLflow run using mlflow.start_run().
You can wrap this call in
with mlflow.start_run(); when you do this, the run is ended automatically after it completes.
- Use MLflow Tracking methods, such as mlflow.log_param(), to track pre-training content.
- Train one or more machine learning models in a framework supported by Databricks Autologging.
- Use MLflow Tracking methods, such as mlflow.log_metric(), to track post-training content.
- If you did not use
with mlflow.start_run()in Step 2, end the MLflow run using mlflow.end_run().
import mlflow mlflow.autolog(exclusive=False) with mlflow.start_run(): mlflow.log_param("example_param", "example_value") # <your model training code here> mlflow.log_param("example_metric", 5)
To disable Databricks Autologging in a Databricks interactive Python notebook, call
import mlflow mlflow.autolog(disable=True)
Administrators can also disable Databricks Autologging for all clusters in a workspace from the Advanced tab of the admin console. Clusters must be restarted for this change to take effect.
Databricks Autologging is supported in interactive Python notebooks and is available for the following ML frameworks:
- Apache Spark MLlib
- PyTorch Lightning
- Fast.ai (version 1.x)
For more information about each of the supported frameworks, see MLflow automatic logging.
All model training information tracked with Databricks Autologging is stored in MLflow Tracking and is secured by MLflow Experiment permissions. You can share, modify, or delete model training information using the MLflow Tracking API or UI.
Administrators can enable or disable Databricks Autologging for all interactive notebook sessions across their workspace in the Advanced tab of the Admin Console. Changes do not take effect until the cluster is restarted.
- Databricks Autologging is not supported in Databricks Runtime 8.4 ML or below, nor in any version of Databricks Runtime. To use autologging in these runtime versions, you can explicitly call mlflow.autolog().
- Databricks Autologging is not supported in Databricks jobs. To use autologging from jobs, you can explicitly call mlflow.autolog().
- Databricks Autologging is enabled only on the driver node of your Databricks cluster. To use autologging from worker nodes, you must explicitly call mlflow.autolog() from within the code executing on each worker.
- The XGBoost scikit-learn integration is not supported.
Disabling the automated MLflow tracking integration for Apache Spark MLlib
TrainValidationSplit models also disables the Databricks Autologging feature for all Apache Spark MLlib models.