machine learning – How to build a lift chart (a.k.a gains chart) in Python?

machine learning – How to build a lift chart (a.k.a gains chart) in Python?

You can use the scikit-plot package to do the heavy lifting.

skplt.metrics.plot_cumulative_gain(y_test, predicted_probas)


# The usual train-test split mumbo-jumbo
from sklearn.datasets import load_breast_cancer
from sklearn.model_selection import train_test_split
from sklearn.naive_bayes import GaussianNB

X, y = load_breast_cancer(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, 
nb = GaussianNB(), y_train)
predicted_probas = nb.predict_proba(X_test)

# The magic happens here
import matplotlib.pyplot as plt
import scikitplot as skplt
skplt.metrics.plot_cumulative_gain(y_test, predicted_probas)

This should result in a plot like this:

Lift/cumulative gains charts arent a good way to evaluate a model (as it cannot be used for comparison between models), and are instead a means of evaluating the results where your resources are finite. Either because theres a cost to action each result (in a marketing scenario) or you want to ignore a certain number of guaranteed voters, and only action those that are on the fence. Where your model is very good, and has high classification accuracy for all results, you wont get much lift from ordering your results by confidence.

import sklearn.metrics
import pandas as pd

def calc_cumulative_gains(df: pd.DataFrame, actual_col: str, predicted_col:str, probability_col:str):

    df.sort_values(by=probability_col, ascending=False, inplace=True)

    subset = df[df[predicted_col] == True]

    rows = []
    for group in np.array_split(subset, 10):
        score = sklearn.metrics.accuracy_score(group[actual_col].tolist(),

        rows.append({NumCases: len(group), NumCorrectPredictions: score})

    lift = pd.DataFrame(rows)

    #Cumulative Gains Calculation
    lift[RunningCorrect] = lift[NumCorrectPredictions].cumsum()
    lift[PercentCorrect] = lift.apply(
        lambda x: (100 / lift[NumCorrectPredictions].sum()) * x[RunningCorrect], axis=1)
    lift[CumulativeCorrectBestCase] = lift[NumCases].cumsum()
    lift[PercentCorrectBestCase] = lift[CumulativeCorrectBestCase].apply(
        lambda x: 100 if (100 / lift[NumCorrectPredictions].sum()) * x > 100 else (100 / lift[
            NumCorrectPredictions].sum()) * x)
    lift[AvgCase] = lift[NumCorrectPredictions].sum() / len(lift)
    lift[CumulativeAvgCase] = lift[AvgCase].cumsum()
    lift[PercentAvgCase] = lift[CumulativeAvgCase].apply(
        lambda x: (100 / lift[NumCorrectPredictions].sum()) * x)

    #Lift Chart
    lift[NormalisedPercentAvg] = 1
    lift[NormalisedPercentWithModel] = lift[PercentCorrect] / lift[PercentAvgCase]

    return lift

To plot the cumulative gains chart, you can use this code below.

    import matplotlib.pyplot as plt
    def plot_cumulative_gains(lift: pd.DataFrame):
        fig, ax = plt.subplots()
        handles = []
        handles.append(ax.plot(lift[PercentCorrect], r-, label=Percent Correct Predictions))
        handles.append(ax.plot(lift[PercentCorrectBestCase], g-, label=Best Case (for current model)))
        handles.append(ax.plot(lift[PercentAvgCase], b-, label=Average Case (for current model)))
        ax.set_xlabel(Total Population (%))
        ax.set_ylabel(Number of Respondents (%))
        ax.set_xlim([0, 9])
        ax.set_ylim([10, 100])
        labels = [int((label+1)*10) for label in [float(item.get_text()) for item in ax.get_xticklabels()]]
        fig.legend(handles, labels=[h[0].get_label() for h in handles])

And to visualise lift:

    def plot_lift_chart(lift: pd.DataFrame):
        plt.plot(lift[NormalisedPercentAvg], r-, label=Normalised response rate with no model)
        plt.plot(lift[NormalisedPercentWithModel], g-, label=Normalised response rate with using model)

Result looks like:


I found these websites useful for reference:


I found the MS link somewhat misleading in its descriptions, but the Paul Te Braak link very informative. To answer the comment;

For the cumulative gains chart above, all the calculations are based upon the accuracy for that specific model. As the Paul Te Braak link notes, how can my models prediction accuracy reach 100% (the red line in the chart)? The best case scenario (the green line) is how quickly we can reach the same accuracy that the red line achieves over the course of the whole population (e.g. our optimum cumulative gains scenario). Blue is if we just randomly pick the classification for each sample in the population. So the cumulative gains and lift charts are purely for understanding how that model (and that model only) will give me more impact in a scenario where Im not going to interact with the entire population.

One scenario I have used the cumulative gains chart is for fraud cases, where I want to know how many applications we can essentially ignore or prioritise (because I know that the model predicts them as well as it can) for the top X percent. In that case, for the average model I instead selected the classification from the real unordered dataset (to show how existing applications were being processed, and how – using the model – we could instead prioritise types of application).

So, for comparing models, just stick with ROC/AUC, and once youre happy with the selected model, use the cumulative gains/ lift chart to see how it responds to the data.

machine learning – How to build a lift chart (a.k.a gains chart) in Python?

You can use the kds package for the same.

For Cummulative Gains Plot:

# pip install kds
import kds
kds.metrics.plot_cumulative_gain(y_test, y_prob)


# Load Dataset and train-test split
from sklearn.datasets import load_iris
from sklearn.model_selection import train_test_split
from sklearn import tree

X, y = load_iris(return_X_y=True)
X_train, X_test, y_train, y_test = train_test_split(X, y, 
clf = tree.DecisionTreeClassifier(max_depth=1,random_state=3)
clf =, y_train)
y_prob = clf.predict_proba(X_test)

import kds
kds.metrics.plot_cumulative_gain(y_test, y_prob[:,1])


Wizard curve will provide the best possible curve for the model.

Disclaimer: I am the author of this package

Leave a Reply

Your email address will not be published.