Use a prediciton dataclass
This commit is contained in:
@@ -1,5 +1,14 @@
|
||||
import random
|
||||
from typing import List, Tuple
|
||||
from dataclasses import dataclass
|
||||
|
||||
|
||||
@dataclass
|
||||
class Prediction:
|
||||
predicted_price: float
|
||||
confidence_score: float
|
||||
similar_listings: List[float]
|
||||
|
||||
|
||||
class HousePricePredictor:
|
||||
"""
|
||||
@@ -11,7 +20,7 @@ class HousePricePredictor:
|
||||
# Mock initialization - in reality would load model weights
|
||||
pass
|
||||
|
||||
def predict(self, square_feet: float, bedrooms: int, bathrooms: float) -> Tuple[float, float, List[float]]:
|
||||
def predict(self, square_feet: float, bedrooms: int, bathrooms: float) -> Prediction:
|
||||
"""
|
||||
Mock prediction method that returns:
|
||||
- predicted price
|
||||
@@ -32,4 +41,4 @@ class HousePricePredictor:
|
||||
for _ in range(3)
|
||||
]
|
||||
|
||||
return predicted_price, confidence_score, similar_listings
|
||||
return Prediction(predicted_price, confidence_score, similar_listings)
|
||||
|
||||
Reference in New Issue
Block a user