junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

removed floor height from binning

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2023-07-04 20:13:18 +01:00
parent df8bfd7d02
commit 946e6d092e
1 changed files with 2 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
model_data/analysis/SapModel.py
View file

@ -181,7 +181,7 @@ class SapModel:
return df
bucket_variables = ["number-open-fireplaces", "fixed-lighting-outlets-count", 'extension-count',
'multi-glaze-proportion', 'floor-height']
'multi-glaze-proportion']
remaining_numericals = [x for x in self.NUMERICAL_COLUMNS if x not in bucket_variables]
for col in bucket_variables:
@ -338,8 +338,7 @@ class SapModel:
def fit_model(self):
# Dummy out the categorical variables
binned = ["number-open-fireplaces", "fixed-lighting-outlets-count", 'extension-count', 'multi-glaze-proportion',
'floor-height']
binned = ["number-open-fireplaces", "fixed-lighting-outlets-count", 'extension-count', 'multi-glaze-proportion']
x = pd.get_dummies(self.model_data, columns=self.CATEGORICAL_COLS + binned, drop_first=True)