Neptune-MLflow Integration

Neptune-mlflow is an open source project curated by Neptune team that enables MLflow experiment runs to be hosted in Neptune.

The integration lets you enjoy the best of both worlds: the tracking and reproducibility of MLflow with the organization and collaboration of Neptune.

organize MLflow experiments in Neptune

Check the example project in Neptune: MLflow integration.

Installation

pip install neptune-mlflow

Sync your MLruns with Neptune

  1. Set your NEPTUNE_API_TOKEN:

Linux/IOS:

export NEPTUNE_API_TOKEN='YOUR_NEPTUNE_API_TOKEN'

Windows:

set NEPTUNE_API_TOKEN="YOUR_NEPTUNE_API_TOKEN"

  1. Navigate to the MLflow project in your directory and run:

neptune mlflow --project USER_NAME/PROJECT_NAME

Alternatively you can point to the MLflow project directory:

neptune mlflow /PATH/TO/MLflow_PROJECT --project WORKSPACE_NAME/PROJECT_NAME

That’s it! You can now browse and collaborate on your MLflow runs in Neptune.

Organize and share your MLflow experiments

organize MLflow experiments in Neptune share artifacts logged during MLflow run

Support

If you need assistance, report an issue in Git, or talk to us directly in a support chat.