File size: 406 Bytes
31460c4 a0f3d24 b4fadea | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 | # app\api\schemas.py
# Pydantic input/output schemas
from pydantic import BaseModel
from typing import Dict
class PredictionRequest(BaseModel):
credit_limit: float
age: int
pay_delay_sep: int
pay_delay_aug: int
bill_amt_sep: float
bill_amt_aug: float
pay_amt_sep: float
pay_amt_aug: float
class PredictionResponse(BaseModel):
prediction: int
probability: float
|