hlnicholls commited on
Commit
cbd5645
·
1 Parent(s): 53d0664

Update shap_plots.py

Browse files
Files changed (1) hide show
  1. shap_plots.py +1 -2
shap_plots.py CHANGED
@@ -391,12 +391,11 @@ def shap_summary_plot(shap_values, features=None, feature_names=None, max_displa
391
  pl.gca().spines['top'].set_visible(False)
392
  pl.gca().spines['left'].set_visible(False)
393
  pl.gca().tick_params(color=axis_color, labelcolor=axis_color)
394
- #pl.yticks(range(len(feature_order)), [feature_names[i] for i in feature_order], fontsize=13)
395
  pl.gca().tick_params('y', length=20, width=0.5, which='major')
396
  pl.gca().tick_params('x', labelsize=11)
397
  pl.ylim(-1, len(feature_order))
398
  pl.xlabel(labels['VALUE'], fontsize=13)
399
- pl.ylabel(feature_names, fontsize=13)
400
  pl.tight_layout()
401
  # if show:
402
  # pl.show()
 
391
  pl.gca().spines['top'].set_visible(False)
392
  pl.gca().spines['left'].set_visible(False)
393
  pl.gca().tick_params(color=axis_color, labelcolor=axis_color)
394
+ pl.yticks(range(len(feature_order)), [feature_names[i] for i in feature_order], labels=feature_names, fontsize=13)
395
  pl.gca().tick_params('y', length=20, width=0.5, which='major')
396
  pl.gca().tick_params('x', labelsize=11)
397
  pl.ylim(-1, len(feature_order))
398
  pl.xlabel(labels['VALUE'], fontsize=13)
 
399
  pl.tight_layout()
400
  # if show:
401
  # pl.show()