python计算ks值 代码_python画KS图,求KS值

ks计算公式

ks用来衡量以一定阈值选定二分类模型预测结果集,各分类命中各自组内比重的差值,某一刻阈值使得此差值最大,此刻的差值就是ks值,ks越大代表模型可以更多地命中某类标签,同时尽可能地错判另一类的标签,具体公式如下:

数据输入

输入:predictions, labels,cut_point

predictions: 为每条样本的预测值组成的集合,预测概率在0-1之间

labels: 为每条样本的真实值(0, 1)组成的集合,本例中1是坏客户

cut_point: KS的阈值分割点的数量

数据预览,左列labels,右列predictions

head -4 test_predict_res.txt

0.0 0.831193

0.0 0.088209815

1.0 0.93411493

0.0 0.022157196

python代码实现

import matplotlib.pyplot as plt

import matplotlib

matplotlib.rcParams["font.sans-serif"] = ["SimHei"]

def ks_plot(pr