add appliiance cols for whole home

rewiringamerica · Jan 9, 2025 · 7891cad · 7891cad
1 parent 6e727b9
commit 7891cad
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/datagen.py b/src/datagen.py
@@ -131,6 +131,8 @@ class DataGenerator(tf.keras.utils.Sequence):
         "neighbor_distance_ft",
         "n_occupants",
         "vintage",
+        "has_heat_pump_dryer",
+        "has_induction_range",
         # fuel indicators -- these must be present for post-processing to work!!
         "has_methane_gas_appliance",
         "has_fuel_oil_appliance",