Skip to main content

Load data into MLflow

This guide shows how to load exported data into MLflow.

Before you start​

  • Export your data from Neptune using the Neptune Exporter tool. Start from the migration overview.
  • Get your MLflow tracking URI. You can set it with the environment variable MLFLOW_TRACKING_URI or in the command line with --mlflow-tracking-uri.

Validate your export​

Before loading the data into MLflow, inspect what was exported:

uv run neptune-exporter summary --data-path ./exports/data

This command returns the number of exported projects or runs, attribute types, and the basic step statistics.

Load the data into MLflow​

The following command loads the data into MLflow, using the default data and file paths:

uv run neptune-exporter load \
  --loader mlflow \
  --mlflow-tracking-uri "http://localhost:5000" \
  --data-path ./exports/data \
  --files-path ./exports/files
info

For detailed instructions, see the Neptune Exporter README file on GitHub.