In the MLflowCallback it is not possible to set the Experiment Name.
As a consequence the run is logged to the Default experiment.

I would like to suggest the following feature:

It is possible to set an env. variable called ML_FLOW_CALLBACK_EXPERIMENT_NAME
which then calls mlflow.set_experiment(experiment_name).

What do you think? As always I can provide a PR if wanted.