接着上篇,在对变量进行分箱后,可能需要计算woe值,虽然转化woe会使变量的信息发生一定程度的损失,但woe在做评分卡时会提供很多便利性,当然在做机器学习时不太建议使用woe,最好直接使用分箱后的数据。
下面直接上干货-代码。
CalcWoe函数中,df为分箱后的数据集,Kvar为主键, Yvar为y值,输出两个对象,df_woe为woe数据集,dict_woe为存放每个变量woe的字典。
下面展示一下输出的df_woe数据集:
dict_woe字典如下:
看看其中一个变量的woe值:
上面是训练样本上的woe转化过程,对于测试样本的woe值直接套用训练样本的woe转化模式。代码如下,其中lst_woe为由训练样本得到的woe的字典。
该函数输出一个对象,即套用训练样本woe转化方式的测试样本woe数据集。
领取专属 10元无门槛券
私享最新 技术干货