mertkarabacak commited on
Commit
596ab4a
1 Parent(s): aaabc0e

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -176,8 +176,7 @@ def y1_interpret_xgb(*args):
176
  df = pd.DataFrame([args], columns=x1.columns)
177
  df = df.astype({col: "category" for col in categorical_columns1})
178
  shap_values = y1_explainer_xgb.shap_values(xgb.DMatrix(df, enable_categorical=True))
179
- fig_m = shap.bar_plot(shap_values[0], max_display=15, show=False)
180
- return fig_m
181
 
182
  def y1_interpret_lgb(*args):
183
  df = pd.DataFrame([args], columns=x1_lgb.columns)
 
176
  df = pd.DataFrame([args], columns=x1.columns)
177
  df = df.astype({col: "category" for col in categorical_columns1})
178
  shap_values = y1_explainer_xgb.shap_values(xgb.DMatrix(df, enable_categorical=True))
179
+ return shap.bar_plot(shap_values[0], max_display=15, show=False)
 
180
 
181
  def y1_interpret_lgb(*args):
182
  df = pd.DataFrame([args], columns=x1_lgb.columns)