org.tensorflow.op.core

Class BoostedTreesUpdateEnsembleV2

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.BoostedTreesUpdateEnsembleV2

All Implemented Interfaces:

Op

public final class BoostedTreesUpdateEnsembleV2
extends PrimitiveOp

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

or by starting a new tree.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type

Method and Description

static BoostedTreesUpdateEnsembleV2

create(Scope scope, Operand<?> treeEnsembleHandle, 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) Factory method to create a class wrapping a new BoostedTreesUpdateEnsembleV2 operation.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, op, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static BoostedTreesUpdateEnsembleV2 create(Scope scope,
                                                  Operand<?> treeEnsembleHandle,
                                                  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)

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.

Returns:

a new instance of BoostedTreesUpdateEnsembleV2