Added some additional features

This commit is contained in:
Khalim Conn-Kowlessar 2024-07-02 18:38:55 +01:00
parent b0449b9e90
commit 77aaecf04f

View file

@ -17,6 +17,10 @@ class EnergyConsumptionModel:
"lodgement-year", "lodgement-month", "current-energy-efficiency", "energy-consumption-current",
"heating-cost-current", "total-floor-area", "number-heated-rooms",
"mainheat-description", "main-fuel", "mainheat-energy-eff", "number-habitable-rooms",
"mainheatcont-description", "property-type", "built-form",
# To test
# "hotwater-description" - make a days since lodgment variable?
# A geographic variable
],
"hot_water_kwh": [
"lodgement-year", "lodgement-month", "current-energy-efficiency", "energy-consumption-current",
@ -26,7 +30,7 @@ class EnergyConsumptionModel:
TARGETS = ['heating_kwh', 'hot_water_kwh']
CATEGORICAL_COLUMNS = [
"lodgement-year", "lodgement-month", "main-fuel", "mainheat-description", "number-heated-rooms",
"number-habitable-rooms", "mainheat-energy-eff"
"number-habitable-rooms", "mainheat-energy-eff", "mainheatcont-description", "property-type", "built-form",
]
NUMERICAL_COLUMNS = ["current-energy-efficiency", "energy-consumption-current", "heating-cost-current",
"hot-water-cost-current", "total-floor-area"]