CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Définissez CoffeeQualityInput avec les champs aroma (float), flavor (float) et altitude (int).
  • Indiquez le response_model pour 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 
Modifier et exécuter le code