Spaces:
Running
Running
File size: 2,686 Bytes
8213d34 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | """Pydantic schemas for API request/response validation."""
from typing import List, Optional
from pydantic import BaseModel, Field
class WeatherFeatures(BaseModel):
"""Input features for temperature prediction."""
min_temp: float = Field(..., description="Minimum temperature (Celsius)")
max_temp: float = Field(..., description="Maximum temperature (Celsius)")
global_radiation: float = Field(..., ge=0, description="Global radiation (W/m2)")
sunshine: float = Field(..., ge=0, le=24, description="Hours of sunshine")
cloud_cover: float = Field(..., ge=0, le=10, description="Cloud cover (oktas, 0-10)")
precipitation: float = Field(..., ge=0, description="Precipitation (mm)")
pressure: float = Field(..., gt=0, description="Atmospheric pressure (Pa)")
snow_depth: float = Field(..., ge=0, description="Snow depth (cm)")
class Config:
json_schema_extra = {
"example": {
"min_temp": 5.2,
"max_temp": 12.8,
"global_radiation": 45.0,
"sunshine": 3.5,
"cloud_cover": 6.0,
"precipitation": 0.5,
"pressure": 101325.0,
"snow_depth": 0.0,
}
}
class PredictionResponse(BaseModel):
"""Response containing the predicted mean temperature."""
predicted_mean_temp: float = Field(..., description="Predicted mean temperature (Celsius)")
model_version: str = Field(..., description="Model version/run ID")
class Config:
json_schema_extra = {"example": {"predicted_mean_temp": 9.5, "model_version": "random_forest_v1"}}
class BatchPredictionRequest(BaseModel):
"""Request for batch predictions."""
instances: List[WeatherFeatures] = Field(..., min_length=1, max_length=1000)
class BatchPredictionResponse(BaseModel):
"""Response containing multiple predictions."""
predictions: List[float] = Field(..., description="List of predicted mean temperatures")
model_version: str = Field(..., description="Model version/run ID")
count: int = Field(..., description="Number of predictions made")
class HealthResponse(BaseModel):
"""Health check response."""
status: str = Field(..., description="Service status")
model_loaded: bool = Field(..., description="Whether model is loaded")
model_path: Optional[str] = Field(None, description="Path to loaded model")
class ModelMetrics(BaseModel):
"""Model performance metrics."""
train_rmse: float
train_mae: float
train_r2: float
valid_rmse: float
valid_mae: float
valid_r2: float
test_rmse: float
test_mae: float
test_r2: float
|