- Set of Databricks notebooks to perform MLflow export and import operations.
- Use these notebooks when you want to copy MLflow objects from one Databricks workspace (tracking server) to another.
- In order to copy MLflow objects between workspaces, you will need to set up a shared cloud bucket mounted on each workspace's DBFS.
- The notebooks are generated with the Databricks GitHub version control feature.
Use the Databricks REST API command databricks workspace import_dir to import the notebooks into a workspace.
git clone https://github.com/mlflow/mlflow-export-import
databricks workspace import_dir \
databricks_notebooks \
/Users/[email protected]/tools
Single Notebooks
Copy one MLflow object and control its destination object name.
| Export | Import |
|---|---|
| Export_Run | Import_Run |
| Export_Experiment | Import_Experiment.py |
| Export_Model | Import_Model.py |
Bulk notebooks
Copy multiple MLflow objects. The target object name will be the same as the source object.
| Export | Import |
|---|---|
| Export_Experiments | Import_Experiments |
| Export_Models | Import_Models |
| Export_All | Use Import_Models |