MLflow_rs
This is a client library for experiment tracking with MLflow. Improvements over the official Python library:
- uncommitted changes will be correctly handled to ensure reproducibility
- logs from log compatible loggers can be stored with the experiment results
- experiment code gets notified if the user wants to terminate the experiment which provides the opportunity to e.g. finish the current iteration / save the current state etc.
- compile time configuration
disable_experiment_tracking
disables experiment tracking and removes most of the code which should result in minimal overhead when experiment tracking needs to be disabled temporarily
Usage
[]
= "0.1"
use ;
use Builder;
use ;
use ;
/// Function that executes the experiment
When you want to disable tracking temporarily:
Create the file .cargo/config.toml
and add:
[]
= ["--cfg", "disable_experiment_tracking"]
or run cargo with:
cargo rustc --lib -- --cfg disable_experiment_tracking
or set the RUSTFLAGS environment variable:
RUSTFLAGS="--cfg disable_experiment_tracking" cargo build --lib
other ways can be found here: https://doc.rust-lang.org/cargo/reference/config.html?highlight=rustflags#buildrustflags