java.lang.Object
org.elasticsearch.xpack.core.ml.dataframe.analyses.BoostedTreeParams
All Implemented Interfaces:
Writeable, org.elasticsearch.xcontent.ToXContent, org.elasticsearch.xcontent.ToXContentFragment

public class BoostedTreeParams extends Object implements org.elasticsearch.xcontent.ToXContentFragment, Writeable
Parameters used by both Classification and Regression analyses.
  • Field Details

    • LAMBDA

      public static final org.elasticsearch.xcontent.ParseField LAMBDA
    • GAMMA

      public static final org.elasticsearch.xcontent.ParseField GAMMA
    • ETA

      public static final org.elasticsearch.xcontent.ParseField ETA
    • MAX_TREES

      public static final org.elasticsearch.xcontent.ParseField MAX_TREES
    • FEATURE_BAG_FRACTION

      public static final org.elasticsearch.xcontent.ParseField FEATURE_BAG_FRACTION
    • NUM_TOP_FEATURE_IMPORTANCE_VALUES

      public static final org.elasticsearch.xcontent.ParseField NUM_TOP_FEATURE_IMPORTANCE_VALUES
    • ALPHA

      public static final org.elasticsearch.xcontent.ParseField ALPHA
    • ETA_GROWTH_RATE_PER_TREE

      public static final org.elasticsearch.xcontent.ParseField ETA_GROWTH_RATE_PER_TREE
    • SOFT_TREE_DEPTH_LIMIT

      public static final org.elasticsearch.xcontent.ParseField SOFT_TREE_DEPTH_LIMIT
    • SOFT_TREE_DEPTH_TOLERANCE

      public static final org.elasticsearch.xcontent.ParseField SOFT_TREE_DEPTH_TOLERANCE
    • DOWNSAMPLE_FACTOR

      public static final org.elasticsearch.xcontent.ParseField DOWNSAMPLE_FACTOR
    • MAX_OPTIMIZATION_ROUNDS_PER_HYPERPARAMETER

      public static final org.elasticsearch.xcontent.ParseField MAX_OPTIMIZATION_ROUNDS_PER_HYPERPARAMETER
  • Constructor Details

    • BoostedTreeParams

      public BoostedTreeParams(@Nullable Double lambda, @Nullable Double gamma, @Nullable Double eta, @Nullable Integer maxTrees, @Nullable Double featureBagFraction, @Nullable Integer numTopFeatureImportanceValues, @Nullable Double alpha, @Nullable Double etaGrowthRatePerTree, @Nullable Double softTreeDepthLimit, @Nullable Double softTreeDepthTolerance, @Nullable Double downsampleFactor, @Nullable Integer maxOptimizationRoundsPerHyperparameter)
  • Method Details

    • getLambda

      public Double getLambda()
    • getGamma

      public Double getGamma()
    • getEta

      public Double getEta()
    • getMaxTrees

      public Integer getMaxTrees()
    • getFeatureBagFraction

      public Double getFeatureBagFraction()
    • getNumTopFeatureImportanceValues

      public Integer getNumTopFeatureImportanceValues()
    • getAlpha

      public Double getAlpha()
    • getEtaGrowthRatePerTree

      public Double getEtaGrowthRatePerTree()
    • getSoftTreeDepthLimit

      public Double getSoftTreeDepthLimit()
    • getSoftTreeDepthTolerance

      public Double getSoftTreeDepthTolerance()
    • getDownsampleFactor

      public Double getDownsampleFactor()
    • getMaxOptimizationRoundsPerHyperparameter

      public Integer getMaxOptimizationRoundsPerHyperparameter()
    • writeTo

      public void writeTo(StreamOutput out) throws IOException
      Specified by:
      writeTo in interface Writeable
      Throws:
      IOException
    • toXContent

      public org.elasticsearch.xcontent.XContentBuilder toXContent(org.elasticsearch.xcontent.XContentBuilder builder, org.elasticsearch.xcontent.ToXContent.Params params) throws IOException
      Specified by:
      toXContent in interface org.elasticsearch.xcontent.ToXContent
      Throws:
      IOException
    • equals

      public boolean equals(Object o)
      Overrides:
      equals in class Object
    • hashCode

      public int hashCode()
      Overrides:
      hashCode in class Object
    • builder

      public static BoostedTreeParams.Builder builder()