hex.genmodel.algos.isotonic.IsotonicRegressionMojoModel Maven / Gradle / Ivy

Go to download

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

H2O GenModel

There is a newer version: 3.46.0.6

package hex.genmodel.algos.isotonic;

import hex.genmodel.MojoModel;

public class IsotonicRegressionMojoModel extends MojoModel {

    protected IsotonicCalibrator _isotonic_calibrator;

    public IsotonicRegressionMojoModel(String[] columns, String[][] domains, String responseColumn) {
        super(columns, domains, responseColumn);
    }

    @Override
    public double[] score0(double[] row, double[] preds) {
        preds[0] = _isotonic_calibrator.calibrateP1(row[0]);
        return preds;
    }

}