fivedollarwitch commited on
Commit
54ed50f
·
verified ·
1 Parent(s): c283d52

Update StatisticalBaseModel.py

Browse files

Add New York and Denver to market handling, add default for days on market.

Files changed (1) hide show
  1. 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]: