Tree.Builder (rest-high-level 7.9.0 API)

java.lang.Object

org.elasticsearch.client.ml.inference.trainedmodel.tree.Tree.Builder

Enclosing class:: Tree

public static class Tree.Builder
extends java.lang.Object

Constructor Summary

Constructors

Constructor Description

Builder()

Method Summary

Modifier and Type	Method	Description
`TreeNode.Builder`	`addJunction(int nodeIndex, int featureIndex, boolean isDefaultLeft, double decisionThreshold)`	Add a decision node.
`Tree.Builder`	`addLeaf(int nodeIndex, double value)`	Sets the node at `nodeIndex` to a leaf node.
`Tree.Builder`	`addNode(TreeNode.Builder node)`
`Tree`	`build()`
`Tree.Builder`	`setClassificationLabels(java.util.List<java.lang.String> classificationLabels)`
`Tree.Builder`	`setFeatureNames(java.util.List<java.lang.String> featureNames)`
`Tree.Builder`	`setNodes(java.util.List<TreeNode.Builder> nodes)`
`Tree.Builder`	`setNodes(TreeNode.Builder... nodes)`
`Tree.Builder`	`setTargetType(TargetType targetType)`

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Details
- Builder
  
  public Builder()
Method Details
- setFeatureNames
  
  public Tree.Builder setFeatureNames(java.util.List<java.lang.String> featureNames)
- addNode
  
  public Tree.Builder addNode(TreeNode.Builder node)
- setNodes
  
  public Tree.Builder setNodes(java.util.List<TreeNode.Builder> nodes)
- setNodes
  
  public Tree.Builder setNodes(TreeNode.Builder... nodes)
- setTargetType
  
  public Tree.Builder setTargetType(TargetType targetType)
- setClassificationLabels
  
  public Tree.Builder setClassificationLabels(java.util.List<java.lang.String> classificationLabels)
- addJunction
  
  public TreeNode.Builder addJunction(int nodeIndex, int featureIndex, boolean isDefaultLeft, double decisionThreshold)
  
  Add a decision node. Space for the child nodes is allocated
  
  Parameters:
  
  nodeIndex - Where to place the node. This is either 0 (root) or an existing child node index
  
  featureIndex - The feature index the decision is made on
  
  isDefaultLeft - Default left branch if the feature is missing
  
  decisionThreshold - The decision threshold
  
  Returns:
  
  The created node
- addLeaf
  
  public Tree.Builder addLeaf(int nodeIndex, double value)
  
  Sets the node at nodeIndex to a leaf node.
  
  Parameters:
  
  nodeIndex - The index as allocated by a call to addJunction(int, int, boolean, double)
  
  value - The prediction value
  
  Returns:
  
  this
- build
  
  public Tree build()