From 946e6d092e04ed419ebab821f53a9a7e168ca6d4 Mon Sep 17 00:00:00 2001 From: Khalim Conn-Kowlessar Date: Tue, 4 Jul 2023 20:13:18 +0100 Subject: [PATCH] removed floor height from binning --- model_data/analysis/SapModel.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/model_data/analysis/SapModel.py b/model_data/analysis/SapModel.py index e0911780..3f3d07eb 100644 --- a/model_data/analysis/SapModel.py +++ b/model_data/analysis/SapModel.py @@ -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)