MLModels / src /schemas.py
pcsekhar's picture
Deploy house price predictor
9e03a51 verified
Raw
History Blame Contribute Delete
528 Bytes
from __future__ import annotations
from pydantic import BaseModel, Field
class HouseFeatures(BaseModel):
OverallQual: int = Field(7, ge=1, le=10)
GrLivArea: int = Field(1800, ge=200)
GarageCars: int = Field(2, ge=0, le=5)
TotalBsmtSF: int = Field(1000, ge=0)
FullBath: int = Field(2, ge=0, le=5)
YearBuilt: int = Field(1995, ge=1800, le=2026)
Neighborhood: str = "Somerst"
HouseStyle: str = "2Story"
class PredictionResponse(BaseModel):
predicted_price: float
currency: str = "USD"