if __name__ == "__main__":
    
    # load, split steps
    ...
    
    # create experiment
    my_exp = mlflow.set_experiment("sklearn_experiment")
    
    # start MLflow run
    with mlflow.start_run(experiment_id=my_exp.experiment_id):
    
        # Train model
        ...