Examples with GradientBaseBooster com.alibaba.alink.operator.common.tree.parallelcart.booster.GradientBaseBooster used on opensource projects

Example 1 with GradientBaseBooster

use of com.alibaba.alink.operator.common.tree.parallelcart.booster.GradientBaseBooster in project Alink by alibaba.

the class BoostingTest method testGradient.

@Test
public void testGradient() {
    GradientBaseBooster booster = new GradientBaseBooster(new LogLoss(2.0, 1.0), new double[] { 1.0, 1.0, 1.0 }, new Slice(0, 3));
    booster.boosting(null, new double[] { 1, 1, 0 }, new double[] { 0.5, 0.5, 0.1 });
    Assert.assertArrayEquals(new double[] { 0.3775406687981454, 0.3775406687981454, -0.52497918747894 }, booster.getGradients(), 1e-6);
    Assert.assertArrayEquals(new double[] { 0.1425369565965509, 0.1425369565965509, 0.27560314728604807 }, booster.getGradientsSqr(), 1e-6);
    Assert.assertNull(booster.getHessions());
    Assert.assertArrayEquals(new double[] { 1.0, 1.0, 1.0 }, booster.getWeights(), 1e-6);
}