Package org.elasticsearch.client.ml.inference.trainedmodel.tree

Class TreeNode.Builder

java.lang.Object

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

Enclosing class:

TreeNode

public static class TreeNode.Builder
extends Object

Constructor Summary

Constructors

Constructor	Description
Builder(int nodeIndex)

Method Summary

Modifier and Type	Method	Description
TreeNode	build()
Integer	getLeftChild()
Integer	getRightChild()
TreeNode.Builder	setDefaultLeft(Boolean defaultLeft)
TreeNode.Builder	setLeafValue(List<Double> leafValue)
TreeNode.Builder	setLeftChild(Integer leftChild)
TreeNode.Builder	setNodeIndex(int nodeIndex)
TreeNode.Builder	setNumberSamples(Long numberSamples)
TreeNode.Builder	setOperator(Operator operator)
TreeNode.Builder	setRightChild(Integer rightChild)
TreeNode.Builder	setSplitFeature(Integer splitFeature)
TreeNode.Builder	setSplitGain(Double splitGain)
TreeNode.Builder	setThreshold(Double threshold)

Methods inherited from class java.lang.Object

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

Constructor Details

Builder

public Builder(int nodeIndex)

Method Details

setOperator

public TreeNode.Builder setOperator(Operator operator)

setThreshold

public TreeNode.Builder setThreshold(Double threshold)

setSplitFeature

public TreeNode.Builder setSplitFeature(Integer splitFeature)

setNodeIndex

public TreeNode.Builder setNodeIndex(int nodeIndex)

setSplitGain

public TreeNode.Builder setSplitGain(Double splitGain)

setLeafValue

public TreeNode.Builder setLeafValue(List<Double> leafValue)

setDefaultLeft

public TreeNode.Builder setDefaultLeft(Boolean defaultLeft)

setLeftChild

public TreeNode.Builder setLeftChild(Integer leftChild)

getLeftChild

public Integer getLeftChild()

setRightChild

public TreeNode.Builder setRightChild(Integer rightChild)

getRightChild

public Integer getRightChild()

setNumberSamples

public TreeNode.Builder setNumberSamples(Long numberSamples)

build

public TreeNode build()