1. Learn
    2. /
  2. Courses
    3. /
  3. Deploying AI into Production with FastAPI
Connected

Exercise

Create a Pydantic model for ML input

You're developing a FastAPI application to deploy a machine learning model that predicts the quality score of coffee based on attributes including aroma, flavor, and altitude.

The first step is to create a Pydantic model to validate the input request data for your ML model and ensure that only valid data flows through the model for successful model prediction.

Instructions

100 XP
  • Import the base validation class from Pydantic to create a data model.
  • Define a class named CoffeeQualityInput that inherits from the Pydantic base class.
  • Add three attributes to the class: aroma (float), flavor (float), and altitude (int).