mlflow example

mlflow_tracking.py

import os
from mlflow import log_metric, log_param, log_artifact
import mlflow

mlflow.set_tracking_uri("http://localhost:5000")
mlflow.set_experiment("my-experiment")

if __name__ == "__main__":
    # Log a parameter (key-value pair)
    log_param("param1", 5)

    # Log a metric; metrics can be updated throughout the run
    log_metric("foo", 1)
    log_metric("foo", 2)
    log_metric("foo", 3)

    # Log an artifact (output file)
    with open("output.txt", "w") as f:
        f.write("Hello world!")
    log_artifact("output.txt")

start-server-and-run.sh

#!/usr/bin/env bash
mlflow server > log 2>&1 &
sleep 5
python mlflow_tracking.py