fix categorical

app.py

@@ -84,13 +84,26 @@ def normalizeData(df):
 
     return df
 
+def xgbFix(df):
+    # XGBoost doesn't like categorical features to be one-hot encoded, so we have to do it manually
+    features_to_categorical = ["area", "streetName", "brf", "agency"]
+
+
+    features_to_float = ["number", "sqm", "rooms", "price", "monthlyFee",
+                        "monthlyCost", "floor", "yearBuilt", "gdp", "unemployment",
+                        "interestRate", "lat", "lon", "soldDate"]
+
+
+    df[features_to_categorical] = df[features_to_categorical].astype("category")
+    df[features_to_float] = df[features_to_float].astype(float)
+    return df
 
 model = downloadModel()
 
 def sthlm(streetName, area, number, sqm, rooms, monthlyFee, monthlyCost, floor, yearBuilt):
     soldDate = '2021-01-01' # TODO
     price = None
-    agency = None
+    agency = 'Notar'
     brf = 'BRF Kartboken 1' # TODO: remove
     lat, lon = getAddressInfo(streetName, number)
     gdp, unemployment, interestRate = getFinancialInfo(soldDate)
@@ -98,9 +111,10 @@ def sthlm(streetName, area, number, sqm, rooms, monthlyFee, monthlyCost, floor,
     # Parse the input so we can run it through the model
     # Create a dataframe from the input values
     input_variables = pd.DataFrame(
-        [[area,streetName,number,sqm,rooms,soldDate,monthlyFee,monthlyCost,floor,yearBuilt, brf,agency,lat,lon,gdp,unemployment,interestRate]], columns=columnHeaders)
+        [[area,streetName,number,sqm,rooms,soldDate,monthlyFee,monthlyCost,floor,yearBuilt,brf,agency,lat,lon,gdp,unemployment,interestRate]], columns=columnHeaders)
     
     df = normalizeData(input_variables)
+    df = xgbFix(df)
 
     # Save first row as a numpy array
     input_list = df.iloc[0].to_numpy()
@@ -118,7 +132,7 @@ numericalInputs = ['number', 'sqm','rooms', 'monthlyFee','monthlyCost','floor','
 categoricalInputs = ['area']
 inputs = [gr.inputs.Textbox(lines=1, label='streetName')]
 catToInput = {
-    'feature': ['a', 'b', 'c']
+    'feature': ['Bromma', 'Abrahamsberg', 'Akalla']
 }
     
 # Generate the input form