MLFlow connection added

This commit is contained in:
Varyngoth
2025-12-04 17:40:39 -04:00
parent c5fb865583
commit 3eef3c7ed5
3 changed files with 51 additions and 1 deletions

View File

@@ -15,7 +15,6 @@ from src.preprocessing import build_preprocessor
from src.models import make_model
from src.stability import compute_shap_matrix
def save_checkpoint(path, gen, pop, seed):
state = {
"gen": gen,

45
runs.txt Normal file
View File

@@ -0,0 +1,45 @@
Cross validation baseline:
python3 -m src.protocols_methodology.exp_cv --dataset adult --algo rf --n-folds 5 --seeds 0 1 2 3 4
Bootstrap baseline:
python3 -m src.protocols_methodology.exp_bootstrap --dataset adult --algo rf --n-bootstrap 30 --seeds 0 1 2 3 4
Noise baseline:
python3 -m src.protocols_methodology.exp_noise --dataset adult --algo rf --n-replicates 30 --noise-std 0.01 --seeds 0 1 2 3 4
Small run: compute 0
tmux new -s small
python3 -m src.protocols_methodology.run_nsga_protocols --dataset adult --generations 3 --pop-size 8 --seed 0
tmux attach-session -t small
Medium run: compute 1
tmux new -s medium
python3 -m src.protocols_methodology.run_nsga_protocols --dataset adult --generations 10 --pop-size 24 --seed 0
tmux attach-session -t medium
Larger run: compute 3
tmux new -s larger
python3 -m src.protocols_methodology.run_nsga_protocols --dataset adult --generations 20 --pop-size 40 --seed 0
tmux attach-session -t larger
Send back:
The three baseline summary csv files:
runs/protocol_cv/__cv_summary.csv
runs/protocol_bootstrap/__bootstrap_summary.csv
runs/protocol_noise/__noise_summary.csv
The three AutoML Pareto fronts:
runs/_protocol_study/cv/pareto_front.csv
runs/_protocol_study/bootstrap/pareto_front.csv
runs/_protocol_study/noise/pareto_front.csv

View File

@@ -18,6 +18,12 @@ from src.protocols_methodology.automl_protocol_adapters import (
noise_protocol,
)
# Main network
# mlflow.set_tracking_uri("http://192.168.2.169:5000")
# Cluster Subnet
mlflow.set_tracking_uri("http://10.10.0.5:5000")
# Network with DNS resolution (specified hosts or Tailnet)
#mlflow.set_tracking_uri("http://medea:5000")
def save_checkpoint(path, gen, pop, seed):
state = {