Skip to content

A beginner's project on automating the training, evaluation, versioning, and deployment of models using GitHub Actions.

License

Notifications You must be signed in to change notification settings

kingabzpro/CICD-for-Machine-Learning

Repository files navigation

CICD-for-Machine-Learning

CI Continuous Deployment Open In Colab

DataCamp Open in Spaces

Learn how to automate model training, evaluation, versioning, and deployment using GitHub Actions with the easiest MLOps guide available online.

Project Description

In this project, we will be using scikit-learn pipelines to train our random forest algorithm and build a drug classifier. After training, we will automate the evaluation process using CML. Finally, we will build and deploy the web application to Hugging Face Hub.

From training to evaluation, the entire process will be automated using GitHub actions. All you have to do is push the code to your GitHub repository, and within two minutes, the model will be updated on Hugging Face with the updated app, model, and results.

Follow the tutorial: https://www.datacamp.com/tutorial/ci-cd-for-machine-learning

Pipeline

CICD

Results

Model Accuracy F1 Score
RandomForestClassifier 97.0% 94.0%

CM