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", "lodgement-year", "lodgement-month", "current-energy-efficiency", "energy-consumption-current",
"heating-cost-current", "total-floor-area", "number-heated-rooms", "heating-cost-current", "total-floor-area", "number-heated-rooms",
"mainheat-description", "main-fuel", "mainheat-energy-eff", "number-habitable-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": [ "hot_water_kwh": [
"lodgement-year", "lodgement-month", "current-energy-efficiency", "energy-consumption-current", "lodgement-year", "lodgement-month", "current-energy-efficiency", "energy-consumption-current",
@ -26,7 +30,7 @@ class EnergyConsumptionModel:
TARGETS = ['heating_kwh', 'hot_water_kwh'] TARGETS = ['heating_kwh', 'hot_water_kwh']
CATEGORICAL_COLUMNS = [ CATEGORICAL_COLUMNS = [
"lodgement-year", "lodgement-month", "main-fuel", "mainheat-description", "number-heated-rooms", "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", NUMERICAL_COLUMNS = ["current-energy-efficiency", "energy-consumption-current", "heating-cost-current",
"hot-water-cost-current", "total-floor-area"] "hot-water-cost-current", "total-floor-area"]