Update StatisticalBaseModel.py
Browse filesAdd New York and Denver to market handling, add default for days on market.
- StatisticalBaseModel.py +9 -0
StatisticalBaseModel.py
CHANGED
|
@@ -59,6 +59,11 @@ class StatisticalBaseModel:
|
|
| 59 |
return 73
|
| 60 |
elif market == 'Chicago':
|
| 61 |
return 25
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 62 |
|
| 63 |
|
| 64 |
def _sale_date_predictor(self, input_data: ProcessedSynapse):
|
|
@@ -103,6 +108,10 @@ class StatisticalBaseModel:
|
|
| 103 |
return 1.15 # 15% above listing
|
| 104 |
elif market == 'Chicago':
|
| 105 |
return 1.12 # 12% above listing
|
|
|
|
|
|
|
|
|
|
|
|
|
| 106 |
return 1.0
|
| 107 |
|
| 108 |
def run_inference(self, input_data: ProcessedSynapse) -> Tuple[float, str]:
|
|
|
|
| 59 |
return 73
|
| 60 |
elif market == 'Chicago':
|
| 61 |
return 25
|
| 62 |
+
elif market == 'New York':
|
| 63 |
+
return 20
|
| 64 |
+
elif market == 'Denver':
|
| 65 |
+
return 24
|
| 66 |
+
return 34
|
| 67 |
|
| 68 |
|
| 69 |
def _sale_date_predictor(self, input_data: ProcessedSynapse):
|
|
|
|
| 108 |
return 1.15 # 15% above listing
|
| 109 |
elif market == 'Chicago':
|
| 110 |
return 1.12 # 12% above listing
|
| 111 |
+
elif market == 'New York':
|
| 112 |
+
return 1.05 # 5% above listing
|
| 113 |
+
elif market == 'Denver':
|
| 114 |
+
return 1.11 ## 11% above listing
|
| 115 |
return 1.0
|
| 116 |
|
| 117 |
def run_inference(self, input_data: ProcessedSynapse) -> Tuple[float, str]:
|