matq007 commited on
Commit
cdb5bcf
·
unverified ·
1 Parent(s): f3ed7d6

fix: SHAP values missing feature column

Browse files
Files changed (1) hide show
  1. pages/3_SHAP_features.py +6 -2
pages/3_SHAP_features.py CHANGED
@@ -1,6 +1,5 @@
1
  #!/usr/bin/env python
2
  import streamlit as st
3
-
4
  from utils import fetch_resource, ui_model_selection
5
 
6
  st.set_page_config(layout="wide")
@@ -37,7 +36,12 @@ if st.session_state["SPECIE"] and st.session_state["VERSION"]:
37
  )
38
 
39
  if explainer:
40
- shap_values = adata.uns["explainer"][explainer].pop("shap_values").reset_index()
 
 
 
 
 
41
  params = [f"{k}:\t{v}" for k, v in adata.uns["explainer"][explainer].items()]
42
 
43
  st.sidebar.markdown("**Parameters**")
 
1
  #!/usr/bin/env python
2
  import streamlit as st
 
3
  from utils import fetch_resource, ui_model_selection
4
 
5
  st.set_page_config(layout="wide")
 
36
  )
37
 
38
  if explainer:
39
+ shap_values = (
40
+ adata.uns["explainer"][explainer]
41
+ .pop("shap_values")
42
+ .reset_index()
43
+ .rename(columns={"index": "feature"})
44
+ )
45
  params = [f"{k}:\t{v}" for k, v in adata.uns["explainer"][explainer].items()]
46
 
47
  st.sidebar.markdown("**Parameters**")