junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactored the model recommendations dataset

Browse source
This commit is contained in:
Michael Duong 2024-01-16 16:41:01 +00:00
parent 61dd26f23e
commit b82fa4cef7
1 changed files with 16 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
recommendations/recommendation_utils.py
View file

@ -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):
"""