From b82fa4cef76bd4b793923db4da87c7036403541d Mon Sep 17 00:00:00 2001 From: Michael Duong Date: Tue, 16 Jan 2024 16:41:01 +0000 Subject: [PATCH] refactored the model recommendations dataset --- recommendations/recommendation_utils.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/recommendations/recommendation_utils.py b/recommendations/recommendation_utils.py index 6244912d..01ef8480 100644 --- a/recommendations/recommendation_utils.py +++ b/recommendations/recommendation_utils.py @@ -310,6 +310,22 @@ def get_roof_u_value( return float(u_value) +def estimate_number_of_floors(property_type): + """ + Using the property type, we estimate the number of floors in the property + """ + + if property_type == "House": + number_of_floors = 2 + elif property_type in ["Flat", "Bungalow"]: + number_of_floors = 1 + elif property_type == "Maisonette": + number_of_floors = 2 + else: + raise NotImplementedError("Implement me") + + return number_of_floors + def estimate_perimeter(floor_area, num_rooms): """