sklearn_fastapi icon indicating copy to clipboard operation
sklearn_fastapi copied to clipboard

Building an API with the FastAPI framework to serve a scikit-learn model.

How to train a simple sklearn model, pickle it, and build an API with fastapi.

Deployment to AWS will be covered in the future.

To follow this post, you should be familiar with:

  1. Scikit-learn
  2. Docker
  3. Basics of an API

Training the model