diff --git a/mlflow/.gitignore b/mlflow/.gitignore new file mode 100644 index 0000000..52f02db --- /dev/null +++ b/mlflow/.gitignore @@ -0,0 +1,2 @@ +mlruns.sqlite +mlartifacts/ diff --git a/requirements.txt b/requirements.txt index 713ffca..b6173f1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,6 +2,7 @@ bokeh >=3.7.2,<4 ipykernel >=6.30.1,<7 ipympl ~=0.9.6 matplotlib >=3.10.1,<4 +mlflow>=2.16,<2.22 numpy >=2.2.6,<3 pandas >=2.3.1,<3 scipy >=1.15.3,<2 diff --git a/run_mlflow_server.sh b/run_mlflow_server.sh new file mode 100755 index 0000000..03a9ddf --- /dev/null +++ b/run_mlflow_server.sh @@ -0,0 +1,13 @@ +# Must be a relative path to be used in an option for mlflow server. +BACKEND_STORE_DB_PATH="./mlflow/mlruns.sqlite" +DEFAULT_ARTIFACTS_ROOT="./mlflow/" + +if [ ! -e "$BACKEND_STORE_DB_PATH" ]; then + printf '%s\n' "Error: '$BACKEND_STORE_DB_PATH' does not exist." >&2 + exit 1 +fi + +mlflow server \ + --backend-store-uri="sqlite:///$BACKEND_STORE_DB_PATH" \ + --default-artifacts-root="$DEFAULT_ARTIFACTS_ROOT" \ + -p 5000