Neptune - MLflow integration 🧩 Experiment tracking with advanced UI, collaborative features, and user access management.
Neptune is a lightweight experiment tracker that offers a single place to track, compare, store, and collaborate on experiments and models.
This integration lets you enjoy tracking and reproducibility of MLflow with the organization and collaboration of Neptune. You can have your MLflow experiment runs hosted in a knowledge repo where you can invite and manage project contributors, while not having to change your MLflow logging code.
Should you wish to switch to Neptune, you can migrate your MLflow data to Neptune with the exporter tool.
On the command line:
pip install neptune-mlflow
Send your MLflow-logged metadata to Neptune (in Python):
import mlflow
from neptune_mlflow_plugin import create_neptune_tracking_uri
# Create a Neptune tracking URI
neptune_uri = create_neptune_tracking_uri(
api_token=ANONYMOUS_API_TOKEN, # Set as environment variable or replace with your own token
project="common/mlflow-integration", # Set as environment variable or replace with your own project
tags=["mlflow", "plugin"], # (optional) use your own
)
mlflow.set_tracking_uri(neptune_uri)
with mlflow.start_run():
...
Export existing MLflow runs to Neptune:
neptune mlflow --project your-neptune-workspace/your-neptune-project
If you got stuck or simply want to talk to us, here are your options: