Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add classes_to_plot option to plot_cumulative_gain #95

Open
etabeta78 opened this issue Oct 23, 2018 · 3 comments
Open

add classes_to_plot option to plot_cumulative_gain #95

etabeta78 opened this issue Oct 23, 2018 · 3 comments

Comments

@etabeta78
Copy link

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

@reiinakano
Copy link
Owner

I guess this makes sense. Open to PRs

@lugq1990
Copy link
Contributor

lugq1990 commented Dec 4, 2018

@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
Copy link
Contributor

lugq1990 commented Dec 4, 2018

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.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

3 participants