Class Tree.Builder
- 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 atnodeIndex
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)
-
-
-
Method Detail
-
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 indexfeatureIndex
- The feature index the decision is made onisDefaultLeft
- Default left branch if the feature is missingdecisionThreshold
- The decision threshold- Returns:
- The created node
-
addLeaf
public Tree.Builder addLeaf(int nodeIndex, double value)
Sets the node atnodeIndex
to a leaf node.- Parameters:
nodeIndex
- The index as allocated by a call toaddJunction(int, int, boolean, double)
value
- The prediction value- Returns:
- this
-
build
public Tree build()
-
-