Vérifier la requête et la réponse pour la prédiction ML.
En vous appuyant sur votre expérience en tant que data scientist au sein de l'entreprise de café, vous devez maintenant créer un point de terminaison FastAPI qui valide les requêtes entrantes à l'aide du modèle de validation de données CoffeeQualityInput
et un QualityPrediction
pour la validation des réponses.
Ce point de terminaison acceptera les données relatives au café et renverra une prévision de qualité accompagnée d'un score de confiance.
Le modèle est déjà chargé dans une fonction appelée « predict_quality
» pour cet exercice.
Cet exercice fait partie du cours
Déployer l'IA en production avec FastAPI
Instructions
- Définissez l'
CoffeeQualityInput
avec les champsaroma
(float
),flavor
(float
) etaltitude
(int
). - Veuillez spécifier l'
response_model
pour valider la réponse dans le décorateur de requête POST. - Veuillez spécifier le modèle de données pour valider la demande d'entrée contenant l'
coffee_data
.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
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