Class TreeNode.Builder
- java.lang.Object
-
- org.elasticsearch.client.ml.inference.trainedmodel.tree.TreeNode.Builder
-
- Enclosing class:
- TreeNode
public static class TreeNode.Builder extends java.lang.Object
-
-
Constructor Summary
Constructors Constructor Description Builder(int nodeIndex)
-
Method Summary
Modifier and Type Method Description TreeNodebuild()java.lang.IntegergetLeftChild()java.lang.IntegergetRightChild()TreeNode.BuildersetDefaultLeft(java.lang.Boolean defaultLeft)TreeNode.BuildersetLeafValue(java.lang.Double leafValue)TreeNode.BuildersetLeftChild(java.lang.Integer leftChild)TreeNode.BuildersetNodeIndex(int nodeIndex)TreeNode.BuildersetOperator(Operator operator)TreeNode.BuildersetRightChild(java.lang.Integer rightChild)TreeNode.BuildersetSplitFeature(java.lang.Integer splitFeature)TreeNode.BuildersetSplitGain(java.lang.Double splitGain)TreeNode.BuildersetThreshold(java.lang.Double threshold)
-
-
-
Method Detail
-
setOperator
public TreeNode.Builder setOperator(Operator operator)
-
setThreshold
public TreeNode.Builder setThreshold(java.lang.Double threshold)
-
setSplitFeature
public TreeNode.Builder setSplitFeature(java.lang.Integer splitFeature)
-
setNodeIndex
public TreeNode.Builder setNodeIndex(int nodeIndex)
-
setSplitGain
public TreeNode.Builder setSplitGain(java.lang.Double splitGain)
-
setLeafValue
public TreeNode.Builder setLeafValue(java.lang.Double leafValue)
-
setDefaultLeft
public TreeNode.Builder setDefaultLeft(java.lang.Boolean defaultLeft)
-
setLeftChild
public TreeNode.Builder setLeftChild(java.lang.Integer leftChild)
-
getLeftChild
public java.lang.Integer getLeftChild()
-
setRightChild
public TreeNode.Builder setRightChild(java.lang.Integer rightChild)
-
getRightChild
public java.lang.Integer getRightChild()
-
build
public TreeNode build()
-
-