Valider la requête et la réponse pour une prédiction ML
Dans le prolongement de votre travail comme data scientist pour l'entreprise de café, vous devez maintenant créer un point de terminaison FastAPI qui valide la requête d'entrée à l'aide du modèle de validation de données CoffeeQualityInput et la réponse à l'aide de QualityPrediction.
Ce point de terminaison acceptera des données sur le café et retournera une prédiction de qualité ainsi qu'un score de confiance.
Le modèle est déjà chargé dans une fonction appelée predict_quality pour cet exercice.
Cette activité fait partie du cours
Déployer l'IA en production avec FastAPI
Instructions de l’exercice
- Définissez
CoffeeQualityInputavec les champsaroma(float),flavor(float) etaltitude(int). - Indiquez le
response_modelpour valider la réponse dans le décorateur de la requête POST. - Indiquez le modèle de données pour valider la requête d'entrée contenant
coffee_data.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
class CoffeeQualityInput(BaseModel):
____: ____
____: ____
____: ____
class QualityPrediction(BaseModel):
quality_score: float
confidence: float
# Specify the data model to validate response
@app.post("/predict", response_model=____)
# Specify the data model to validate input request
def predict(coffee_data: ____):
prediction = predict_quality(coffee_data)
return prediction