hex.tree.uplift.KLDivergence Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-algos Show documentation

H2O Algorithms

There is a newer version: 3.46.0.6

package hex.tree.uplift;

import static water.util.MathUtils.log2;

public class KLDivergence extends Divergence {
    
    @Override
    public double metric(double prCT1, double prCT0) {
        return prCT1 * log2(prCT1 / prCT0 == 0 ? ZERO_TO_DIVIDE : prCT0);
    }
    
    @Override
    public double norm(
        double prCT1, double prCT0, 
        double prLCT1, double prLCT0
    ) {
        double klCT = node(prCT1, prCT0);
        double entCT =  -(prCT1 * log2(prCT1) + prCT0 * log2(prCT0));
        double entCT1 = -(prLCT1 * log2(prLCT1) + (1 - prLCT1) * log2((1 - prLCT1)));
        double entCT0 = -(prLCT0 * log2(prLCT0) + (1 - prLCT0) * log2((1 - prLCT0)));
        return klCT * entCT + prCT1 * entCT1 + prCT0 * entCT0 + 0.5;
    }
    
}