BoostedTreesUpdateEnsembleV2

public final class BoostedTreesUpdateEnsembleV2

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

or by starting a new tree.

Nested Classes

class BoostedTreesUpdateEnsembleV2.Options Optional attributes for BoostedTreesUpdateEnsembleV2  

Public Methods

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

Inherited Methods

Public Methods

public static BoostedTreesUpdateEnsembleV2 create (Scope scope, Operand<?> treeEnsembleHandle, Iterable<Operand<Integer>> featureIds, Iterable<Operand<Integer>> dimensionIds, Iterable<Operand<Integer>> nodeIds, Iterable<Operand<Float>> gains, Iterable<Operand<Integer>> thresholds, Iterable<Operand<Float>> leftNodeContribs, Iterable<Operand<Float>> rightNodeContribs, Iterable<Operand<String>> splitTypes, Operand<Integer> maxDepth, Operand<Float> learningRate, Operand<Integer> pruningMode, Options... options)

Factory method to create a class wrapping a new BoostedTreesUpdateEnsembleV2 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.
dimensionIds List of rank 1 tensors representing the dimension in each feature.
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.
splitTypes List of rank 1 tensors representing the split type for each feature.
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.
options carries optional attributes values
Returns
  • a new instance of BoostedTreesUpdateEnsembleV2

public static BoostedTreesUpdateEnsembleV2.Options logitsDimension (Long logitsDimension)

Parameters
logitsDimension scalar, dimension of the logits