orangewise / scikit-learn-tree-py

Running a decision tree model with mlflow.

development setup

# create virtual env
python -m venv .venv

# activate virtual env
source .venv/bin/activate

# install deps
pip install -r requirements.txt

development run

$ python tree.py <max_depth> <criterion>

mlflow run

# run without mlruns.db if you don't want to use the registry
# export MLFLOW_TRACKING_URI=sqlite:///mlruns.db
mlflow run .

check results

# run without mlruns.db if you don't want to use the registry
mlflow ui --backend-store-uri sqlite:///mlruns.db