File size: 154 Bytes
3d6943b
 
 
 
 
 
1
2
3
4
5
6
import mlflow


def setup_mlflow(experiment_name: str):
    mlflow.set_tracking_uri("sqlite:///mlflow.db")
    mlflow.set_experiment(experiment_name)