DEV Community

Julien Simon for AWS

Posted on • Originally published at Medium on

Deploying Machine Learning Models with mlflow and Amazon SageMaker

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/

Top comments (1)

Collapse
 
swati2901 profile image
swati2901

Hi Julien,

I have tried mlflow on windows10 but ui is not opening.