Update dynamic_shap_plot.py

dynamic_shap_plot.py

@@ -65,17 +65,16 @@ def summary_plot_plotly_fig(shap_values, df_shap, feature_names, max_display = 8
 
     feature_order = np.argsort(np.sum(np.abs(shap_values), axis=0)[:-1])
     feature_order = feature_order[-min(max_display, len(feature_order)):]
-    text = [feature_names[i] for i in feature_order]
+    text = [df_shap.index[i] for i in df_shap.index]
     text = iter(text)
 
     for i in range(1, len(plotly_fig['data']), 2):
         t = text.__next__()
         plotly_fig['data'][i]['name'] = ''
-        #plotly_fig['data'][i]['text'] = t
+        plotly_fig['data'][i]['text'] = t
         plotly_fig['data'][i]['hoverinfo'] = 'text'
-        plotly_fig['data'][i]['text'] = df_shap.index
-        plotly_fig['data'][i]['y'] = [t] * len(df_shap.index)
-        #plotly_fig['data'][i]['y'] = feature_names[feature_order] 
+        #plotly_fig['data'][i]['text'] = df_shap.index
+        plotly_fig['data'][i]['y'] = feature_names[feature_order] 
         
 
     colorbar_trace  = go.Scatter(x=[None],
@@ -104,7 +103,7 @@ def summary_plot_plotly_fig(shap_values, df_shap, feature_names, max_display = 8
                                   x=1.18,
                                   align="right",
                                   valign="top",
-                                  text='Feature value',
+                                  text='Gene',
                                   showarrow=False,
                                   xref="paper",
                                   yref="paper",