MLFlow pseudocode example

mlflow.py

import mlflow

with mlflow.start_run():
    model = ... # define model here
    
    mlflow.log_params(config)   # log hyperparameter dictionary
    
    for epoch_idx in range(num_epochs):
        train_loss = train(model, data)
        val_loss = validate(model, data)
       
        mlflow.log_param("train_loss", train_loss)  # log training loss
        mlflow.log_param("val_loss", val_loss)      # log validation loss