Training and deploying with XGBoost and MLflow