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 TreeNode
build()
java.lang.Integer
getLeftChild()
java.lang.Integer
getRightChild()
TreeNode.Builder
setDefaultLeft(java.lang.Boolean defaultLeft)
TreeNode.Builder
setLeafValue(java.lang.Double leafValue)
TreeNode.Builder
setLeftChild(java.lang.Integer leftChild)
TreeNode.Builder
setNodeIndex(int nodeIndex)
TreeNode.Builder
setOperator(Operator operator)
TreeNode.Builder
setRightChild(java.lang.Integer rightChild)
TreeNode.Builder
setSplitFeature(java.lang.Integer splitFeature)
TreeNode.Builder
setSplitGain(java.lang.Double splitGain)
TreeNode.Builder
setThreshold(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()
-
-