From 8b070e8cf41c269820d910a04dcdd3aee5c0e1d7 Mon Sep 17 00:00:00 2001 From: jwaton Date: Fri, 21 Dec 2018 14:27:45 +0000 Subject: [PATCH] Add condition to only plot non-highlighted paths in the case there are some to plot --- stability_selection/stability_selection.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/stability_selection/stability_selection.py b/stability_selection/stability_selection.py index f0d8ab8..0c7ea97 100644 --- a/stability_selection/stability_selection.py +++ b/stability_selection/stability_selection.py @@ -136,8 +136,9 @@ def plot_stability_path(stability_selection, threshold_highlight=None, x_grid = stability_selection.lambda_grid / np.max(stability_selection.lambda_grid) fig, ax = plt.subplots(1, 1, **kwargs) - ax.plot(x_grid, stability_selection.stability_scores_[~paths_to_highlight].T, - 'k:', linewidth=0.5) + if not paths_to_highlight.all(): + ax.plot(x_grid, stability_selection.stability_scores_[~paths_to_highlight].T, + 'k:', linewidth=0.5) if paths_to_highlight.any(): ax.plot(x_grid, stability_selection.stability_scores_[paths_to_highlight].T,