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

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
hashCode()
Operation
op()
Returns the underlying Operation
final String
toString()
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

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