Mpodszus commited on
Commit
cd9614e
·
verified ·
1 Parent(s): 6aebb83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -123,7 +123,8 @@ def main_func(Inventory_Bucket, Category, Product, Geography, Conference, Deal_S
123
  new_row = new_row[expected_columns]
124
 
125
  dtest = xgb.DMatrix(new_row)
126
- prediction = loaded_model.predict(dtest)[0]
 
127
  shap_values = explainer(dtest)
128
 
129
  fig, ax = plt.subplots(figsize=(8, 4))
 
123
  new_row = new_row[expected_columns]
124
 
125
  dtest = xgb.DMatrix(new_row)
126
+ log_prediction = loaded_model.predict(dtest)[0]
127
+ prediction = np.expm1(log_prediction)
128
  shap_values = explainer(dtest)
129
 
130
  fig, ax = plt.subplots(figsize=(8, 4))