scikit-plot icon indicating copy to clipboard operation
scikit-plot copied to clipboard

add classes_to_plot option to plot_cumulative_gain

Open etabeta78 opened this issue 7 years ago • 3 comments

I think it could be useful, when one wants to plot only e.g. class 1, to have an option to produce consistent plots for both plot_cumulative_gain and plot_roc

At the moment, instead, only plot_roc supports such option.

Thanks a lot

etabeta78 avatar Oct 23 '18 14:10 etabeta78

I guess this makes sense. Open to PRs

reiinakano avatar Oct 24 '18 09:10 reiinakano

@reiinakano I have changed plot_cumulative_gain for choosing classes for ploting also with 'micro' and 'macro' as default ploting curve. You can check it out.

lugq1990 avatar Dec 04 '18 08:12 lugq1990

Because pytest of unit test for plot_cumulative_plot is not allowed for multi-classes plot originally, so this may cause pytest can't pass.

lugq1990 avatar Dec 04 '18 11:12 lugq1990