in sagemaker_notebook_instance/containers/dashboard/src/pages/global_page.py [0:0]
def dependency(explanation_group):
names = extract_feature_names(explanation_group)
feature_idx = st.selectbox(
label="Select feature:",
options=range(len(names)),
format_func=lambda e: names[e],
index=1
)
plot_placeholder = st.empty()
interaction = interaction_checkbox()
shap_values = [[r['explanation']['shap_values'][e] for e in names] for r in explanation_group]
features = [[r['features'][e] for e in names] for r in explanation_group]
interaction_index = "auto" if interaction else None
shap.dependence_plot(
ind=feature_idx,
shap_values=np.array(shap_values),
features=np.array(features),
feature_names=names,
interaction_index=interaction_index
)
plot_placeholder.pyplot(bbox_inches='tight')
plt.clf()