In this video, I first train an XGBoost model on my local machine (I use PyCharm), and visualize results in the mlflow UI. Then, I deploy the model locally, and predict test data. Next, I create a Docker container, push it to Amazon ECR, and use it to deploy my model on Amazon SageMaker.
Happy to answer any question you may have!
Code: https://gitlab.com/juliensimon/sagemaker-automation/-/tree/master/mlflow/direct-marketing-xgboost
Documentation: https://www.mlflow.org/