21 lines
666 B
Python
21 lines
666 B
Python
from pydantic import BaseModel, Field
|
|
from typing import List
|
|
|
|
class HousePredictionInput(BaseModel):
|
|
square_feet: float = Field(..., gt=0, description="Square footage of the house")
|
|
bedrooms: int = Field(..., ge=1, description="Number of bedrooms")
|
|
bathrooms: float = Field(..., gt=0, description="Number of bathrooms")
|
|
|
|
class Config:
|
|
json_schema_extra = {
|
|
"example": {
|
|
"square_feet": 2000,
|
|
"bedrooms": 3,
|
|
"bathrooms": 2.5
|
|
}
|
|
}
|
|
|
|
class HousePredictionOutput(BaseModel):
|
|
predicted_price: float
|
|
confidence_score: float
|
|
similar_listings: List[float] |