Validar requisição e resposta para predição de ML
Dando continuidade ao seu trabalho como cientista de dados na empresa de café, você agora precisa criar um endpoint FastAPI que valide a requisição de entrada usando o modelo de validação de dados CoffeeQualityInput e utilize QualityPrediction para validação da resposta.
Esse endpoint vai receber dados do café e retornar uma predição de qualidade junto com a pontuação de confiança.
O modelo já está carregado em uma função chamada predict_quality para este exercício.
Este exercício faz parte do curso
Colocando IA em Produção com FastAPI
Instruções do exercício
- Defina
CoffeeQualityInputcom os camposaroma(float),flavor(float) ealtitude(int). - Especifique o
response_modelpara validar a resposta no decorator do POST. - Especifique o modelo de dados para validar a requisição de entrada contendo
coffee_data.
Exercício interativo prático
Experimente este exercício completando este código de exemplo.
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