All Downloads are FREE. Search and download functionalities are using the official Maven repository.

hex.schemas.SharedTreeV3 Maven / Gradle / Ivy

package hex.schemas;

import hex.tree.SharedTree;
import hex.tree.SharedTreeModel.SharedTreeParameters;
import water.api.*;

public class SharedTreeV3, P extends SharedTreeV3.SharedTreeParametersV3> extends SupervisedModelBuilderSchema {

  public static class SharedTreeParametersV3

> extends SupervisedModelParametersSchema { static public String[] own_fields = new String[] { "ntrees", "max_depth", "min_rows", "nbins", "r2_stopping", "seed" }; @API(help="Number of trees.", gridable = true) public int ntrees; @API(help="Maximum tree depth.", gridable = true) public int max_depth; @API(help="Fewest allowed observations in a leaf (in R called 'nodesize').", gridable = true) public int min_rows; @API(help="Build a histogram of this many bins, then split at the best point", gridable = true) public int nbins; @API(help="Stop making trees when the r^2 metric equals or exceeds this") public double r2_stopping; @API(help = "Seed for pseudo random number generator (if applicable)", level = API.Level.expert) public long seed; } }





© 2015 - 2025 Weber Informatics LLC | Privacy Policy