org.tensorflow.op.core

Class BoostedTreesUpdateEnsemble

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.BoostedTreesUpdateEnsemble

All Implemented Interfaces:

Op

public final class BoostedTreesUpdateEnsemble
extends PrimitiveOp

Updates the tree ensemble by either 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 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 to wrap a new BoostedTreesUpdateEnsemble operation to the graph.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

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

Method Detail

create

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 to wrap a new BoostedTreesUpdateEnsemble operation to the graph.

Parameters:

scope - current graph 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