hlnicholls commited on
Commit
53f9124
·
1 Parent(s): a62f067

Update dynamic_shap_plot.py

Browse files
Files changed (1) hide show
  1. dynamic_shap_plot.py +3 -1
dynamic_shap_plot.py CHANGED
@@ -71,8 +71,10 @@ def summary_plot_plotly_fig(shap_values, df_shap, feature_names, max_display = 8
71
  for i in range(1, len(plotly_fig['data']), 2):
72
  t = text.__next__()
73
  plotly_fig['data'][i]['name'] = ''
74
- plotly_fig['data'][i]['text'] = t
75
  plotly_fig['data'][i]['hoverinfo'] = 'text'
 
 
76
 
77
  colorbar_trace = go.Scatter(x=[None],
78
  y=[None],
 
71
  for i in range(1, len(plotly_fig['data']), 2):
72
  t = text.__next__()
73
  plotly_fig['data'][i]['name'] = ''
74
+ #plotly_fig['data'][i]['text'] = t
75
  plotly_fig['data'][i]['hoverinfo'] = 'text'
76
+ plotly_fig['data'][i]['text'] = df_shap.index
77
+ plotly_fig['data'][i]['y'] = feature_names[feature_order]
78
 
79
  colorbar_trace = go.Scatter(x=[None],
80
  y=[None],