Skip to content

Latest commit

 

History

History
23 lines (17 loc) · 1.05 KB

README.md

File metadata and controls

23 lines (17 loc) · 1.05 KB

Sentence Classification

This repository contains step by step procedure for using and getting hands practice with MLOps tools. This can be generalized to any machine leaning based project. The goal of this repository is to understand MLOps pipeline and steps. For instance, model buidling, monitoring, configurations, testing, packaging, deployment, and cicd etc etc.

alt text image by ml-ops.org

Repository setup

In the first phase, the model is developed using pytorch lightening framework. The dataset is loaded, preprocessed, and used for training and evaluation. The following steps are covered.

  • Getting the dataset
  • Preprocessing the dataset
  • Define the DataLoaders
  • Define the model
  • Train the model
  • Perform the inference
  • ONNX model conversion

Acknowlegments

  1. https://github.com/graviraja/MLOps-Basics
  2. https://github.com/kelvins/awesome-mlops
  3. https://ml-ops.org/content/mlops-principles