SAS代码 - 试图了解如何构建GINI索引

时间:2016-10-26 23:44:04

标签: python sas gini

我正在参加数据科学竞赛,我的最终预测将通过GINI指数来衡量。这是一个回归问题。我有SAS的计算源代码,但我不知道SAS,我无法理解发生了什么。

我想在Python中构建相同的东西。任何帮助,将不胜感激。如果有人知道Python代码,那将会有很大帮助。

    *define GINI;       
    %macro  gini(input=, output=, y=, py=, filter=, split_ind = );
    data indsn;
        set &input.;
        _random=ranuni (123456789);
        w=1;
        if &split_ind.="&filter.";
    run;

    proc sort data=indsn;by &py _random;run;
        /*accumulate w to calculate Gini    */
        data test;
            set indsn;
            if _N_ = 1 then do;
                cumm_w0=0;
            end;
            retain cumm_w0
            ;
            cumm_w0=cumm_w0+w;
        run;

        /*calcualate Gini */
        proc sql noprint;
            create table &output
            as
            select 1-2/(sum(w)-1)*(sum(w)-sum(&y.*cumm_w0*w)/sum(&y.*w)) as gini
            from test;
        quit;


        proc print data=&output;
            title " GINI on &filter.";run;
    %mend;

1 个答案:

答案 0 :(得分:0)

这看起来像是关于基尼系数的维基百科文章的这一部分的底部公式的实现:

https://en.wikipedia.org/wiki/Gini_coefficient#Alternate_expressions