alinaL commited on
Commit
6b71fc3
1 Parent(s): 4072173

return log price

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -4,6 +4,7 @@ import xgboost
4
  from xgboost import XGBRegressor
5
  import pickle
6
  import pandas as pd
 
7
 
8
  # filename = 'xgboost_model_all.pickle'
9
  # xgb = pickle.load(open(filename, 'rb'))
@@ -34,7 +35,7 @@ def predict_price(area, year, rooms, floor_number, distance_from_center, alarm_s
34
  'details__Ypatybės__Virtuvė sujungta su kambariu': kitchen_plus_livingroom
35
  }])
36
  print(x)
37
- return int(xgb.predict(x))
38
 
39
 
40
  iface = gr.Interface(fn=predict_price, inputs=[gr.inputs.Slider(10, 180, 1, 80),
 
4
  from xgboost import XGBRegressor
5
  import pickle
6
  import pandas as pd
7
+ import numpy as np
8
 
9
  # filename = 'xgboost_model_all.pickle'
10
  # xgb = pickle.load(open(filename, 'rb'))
 
35
  'details__Ypatybės__Virtuvė sujungta su kambariu': kitchen_plus_livingroom
36
  }])
37
  print(x)
38
+ return np.log(int(xgb.predict(x)))
39
 
40
 
41
  iface = gr.Interface(fn=predict_price, inputs=[gr.inputs.Slider(10, 180, 1, 80),