hlnicholls commited on
Commit
be16ea9
1 Parent(s): 1b12e1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -104,7 +104,7 @@ if len(gene_list) > 1:
104
  df_shap.drop(columns='XGB_Score', inplace=True)
105
  shap_values = explainer.shap_values(df_shap)
106
  #summary_plot = shap.summary_plot(shap_values, df_shap)
107
- summary_plot = shap_summary_plot(shap_values, max_display=8)
108
  st.pyplot(fig=summary_plot)
109
  st.caption("SHAP Summary Plot of All Input Genes")
110
  feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
 
104
  df_shap.drop(columns='XGB_Score', inplace=True)
105
  shap_values = explainer.shap_values(df_shap)
106
  #summary_plot = shap.summary_plot(shap_values, df_shap)
107
+ summary_plot = shap_summary_plot(shap_values,feature_names=df.columns, max_display=8)
108
  st.pyplot(fig=summary_plot)
109
  st.caption("SHAP Summary Plot of All Input Genes")
110
  feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])