Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upadd classes_to_plot option to plot_cumulative_gain #95
Labels
Comments
|
I guess this makes sense. Open to PRs |
|
@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. |
|
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
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