mdurst12 commited on
Commit
1bdedaf
·
1 Parent(s): 7fbeebd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def main_func(EAL_SCORE,SOVI_SCORE,RESL_SCORE,DROUGHT_RISKS,EARTHQUAKE_RISKS,HUR
25
  prob = loaded_model.predict_proba(new_row)
26
 
27
  # compute SHAP values
28
- explainer = shap.TreeExplainer(xgb_clf)
29
  shap_values = explainer.shap_values(new_row,check_additivity=False)
30
  # plot = shap.summary_plot(shap_values, new_row.values, plot_type="bar", class_names= class_names, feature_names = new_row.columns)
31
  plot = shap.summary_plot(shap_values[1], new_row.values, feature_names = new_row.columns)
 
25
  prob = loaded_model.predict_proba(new_row)
26
 
27
  # compute SHAP values
28
+ explainer = shap.TreeExplainer(loaded_model)
29
  shap_values = explainer.shap_values(new_row,check_additivity=False)
30
  # plot = shap.summary_plot(shap_values, new_row.values, plot_type="bar", class_names= class_names, feature_names = new_row.columns)
31
  plot = shap.summary_plot(shap_values[1], new_row.values, feature_names = new_row.columns)