Machine Learning on Docker

In this article, we are going to see how we can run a machine learning model in the docker container.

First, we need to create a machine learning model. In this article, we are creating a salary prediction model.

Step1: The first step is to load the dataset

Step2: Split Data into training and testing data.

Step3: Train the model

Step4: To Display Prediction Line/ Best fit line

Step5: Save model for future use for this we are using the joblib module

To run in container.

Step1: Download the image.

docker pull image_name:version 

Step2: Run the container

docker run <options> --name container_name image:version 

Step3: Install python and required library

Step3:copy model to container

Step4: loaf model against with the joblib

Step5: Run the application

congratulation we successfully run the machine learning model in a docker container.

follow me for more interesting articles.

Thank you for reading…