reiinakano/scikit-plot

add classes_to_plot option to plot_cumulative_gain

Open

#95 opened on Oct 23, 2018

View on GitHub
 (3 comments) (0 reactions) (0 assignees)Python (2,352 stars) (282 forks)batch import
easyenhancementgood first issue

Description

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

Contributor guide

Tech stack
python
Domain
datamachine learning
Issue type
feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
2
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
under 1 hour
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
stale
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
clear
Prerequisites
Pythonscikit plotplot roc function
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
70
Research direction
The issue requests adding a 'classes to plot' parameter to the plot cumulative gain function, similar to the existing plot roc function. The relevant code is likely in the scikitplot module. Look at the implementation of plot roc in the codebase to see how 'classes to plot' is handled, and add similar logic to plot cumulative gain. Also update any relevant tests and documentation. Check the existing comments on the issue for any additional guidance.
add classes_to_plot option to plot_cumulative_gain · reiinakano/scikit-plot#95 | Good First Issue