hex.schemas.GBMV3 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.schemas;

import hex.tree.gbm.GBM;
import hex.tree.gbm.GBMModel.GBMParameters;
import water.api.API;


public class GBMV3 extends SharedTreeV3 {

  public static final class GBMParametersV3 extends SharedTreeV3.SharedTreeParametersV3 {
    static public String[] own_fields = new String[] {
        "learn_rate",
        "distribution",
    };

    // Input fields
    @API(help="Learning rate from 0.0 to 1.0", gridable = true)
    public float learn_rate;

    @API(help = "Distribution function", values = { "AUTO", "bernoulli", "multinomial", "gaussian" }, gridable = true)
    public GBMParameters.Family distribution;
  }
}