Update StatisticalBaseModel.py
Browse files- StatisticalBaseModel.py +1 -4
StatisticalBaseModel.py
CHANGED
|
@@ -104,6 +104,7 @@ class StatisticalBaseModel:
|
|
| 104 |
return 1.15 # 15% above listing
|
| 105 |
elif market == 'Chicago':
|
| 106 |
return 1.12 # 12% above listing
|
|
|
|
| 107 |
|
| 108 |
def run_inference(self, input_data: ProcessedSynapse) -> Tuple[float, str]:
|
| 109 |
"""
|
|
@@ -111,12 +112,8 @@ class StatisticalBaseModel:
|
|
| 111 |
:param input_data: a formatted Synapse from the validator, representing a currently listed house
|
| 112 |
:return: the predicted sale price and predicted sale date for this home
|
| 113 |
"""
|
| 114 |
-
print("Running inference")
|
| 115 |
-
print('input data', input_data)
|
| 116 |
listing_price = float(input_data['price']) if 'price' in input_data else 1.0
|
| 117 |
sale_multiplier = self._get_price_multiplier(input_data['market']) if 'market' in input_data else 1.0
|
| 118 |
-
print(f"Listing Price: {listing_price}")
|
| 119 |
-
print(f"Sale Multiplier: {sale_multiplier}")
|
| 120 |
predicted_sale_price = listing_price * sale_multiplier
|
| 121 |
predicted_sale_date = self._sale_date_predictor(input_data)
|
| 122 |
predicted_sale_date = predicted_sale_date.strftime("%Y-%m-%d")
|
|
|
|
| 104 |
return 1.15 # 15% above listing
|
| 105 |
elif market == 'Chicago':
|
| 106 |
return 1.12 # 12% above listing
|
| 107 |
+
return 1.0
|
| 108 |
|
| 109 |
def run_inference(self, input_data: ProcessedSynapse) -> Tuple[float, str]:
|
| 110 |
"""
|
|
|
|
| 112 |
:param input_data: a formatted Synapse from the validator, representing a currently listed house
|
| 113 |
:return: the predicted sale price and predicted sale date for this home
|
| 114 |
"""
|
|
|
|
|
|
|
| 115 |
listing_price = float(input_data['price']) if 'price' in input_data else 1.0
|
| 116 |
sale_multiplier = self._get_price_multiplier(input_data['market']) if 'market' in input_data else 1.0
|
|
|
|
|
|
|
| 117 |
predicted_sale_price = listing_price * sale_multiplier
|
| 118 |
predicted_sale_date = self._sale_date_predictor(input_data)
|
| 119 |
predicted_sale_date = predicted_sale_date.strftime("%Y-%m-%d")
|