BoostedTreesUpdateEnsemble

public final class BoostedTreesUpdateEnsemble

Updates the tree ensemble by either adding a layer to the last tree being grown

or by starting a new tree.

Public Methods

static BoostedTreesUpdateEnsemble
create(Scope scope, Operand<?> treeEnsembleHandle, Operand<Integer> featureIds, Iterable<Operand<Integer>> nodeIds, Iterable<Operand<Float>> gains, Iterable<Operand<Integer>> thresholds, Iterable<Operand<Float>> leftNodeContribs, Iterable<Operand<Float>> rightNodeContribs, Operand<Integer> maxDepth, Operand<Float> learningRate, Long pruningMode)
Factory method to create a class wrapping a new BoostedTreesUpdateEnsemble operation.

Inherited Methods

Public Methods

public static BoostedTreesUpdateEnsemble create (Scope scope, Operand<?> treeEnsembleHandle, Operand<Integer> featureIds, Iterable<Operand<Integer>> nodeIds, Iterable<Operand<Float>> gains, Iterable<Operand<Integer>> thresholds, Iterable<Operand<Float>> leftNodeContribs, Iterable<Operand<Float>> rightNodeContribs, Operand<Integer> maxDepth, Operand<Float> learningRate, Long pruningMode)

Factory method to create a class wrapping a new BoostedTreesUpdateEnsemble operation.

Parameters
scope current scope
treeEnsembleHandle Handle to the ensemble variable.
featureIds Rank 1 tensor with ids for each feature. This is the real id of the feature that will be used in the split.
nodeIds List of rank 1 tensors representing the nodes for which this feature has a split.
gains List of rank 1 tensors representing the gains for each of the feature's split.
thresholds List of rank 1 tensors representing the thesholds for each of the feature's split.
leftNodeContribs List of rank 2 tensors with left leaf contribs for each of the feature's splits. Will be added to the previous node values to constitute the values of the left nodes.
rightNodeContribs List of rank 2 tensors with right leaf contribs for each of the feature's splits. Will be added to the previous node values to constitute the values of the right nodes.
maxDepth Max depth of the tree to build.
learningRate shrinkage const for each new tree.
pruningMode 0-No pruning, 1-Pre-pruning, 2-Post-pruning.
Returns
  • a new instance of BoostedTreesUpdateEnsemble