ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

BoostedTreesGetEnsembleStates

public final class BoostedTreesGetEnsembleStates

Retrieves the tree ensemble resource stamp token, number of trees and growing statistics.

Public Methods

static BoostedTreesGetEnsembleStates
create(Scope scope, Operand<?> treeEnsembleHandle)
Factory method to create a class wrapping a new BoostedTreesGetEnsembleStates operation.
Output<Integer>
lastLayerNodesRange()
Rank size 2 tensor that contains start and end ids of the nodes in the latest layer.
Output<Integer>
numAttemptedLayers()
The number of layers we attempted to build (but not necessarily succeeded).
Output<Integer>
numFinalizedTrees()
The number of trees that were finished successfully.
Output<Integer>
numTrees()
The number of trees in the tree ensemble resource.
Output<Long>
stampToken()
Stamp token of the tree ensemble resource.

Inherited Methods

Public Methods

public static BoostedTreesGetEnsembleStates create (Scope scope, Operand<?> treeEnsembleHandle)

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

Parameters
scope current scope
treeEnsembleHandle Handle to the tree ensemble.
Returns
  • a new instance of BoostedTreesGetEnsembleStates

public Output<Integer> lastLayerNodesRange ()

Rank size 2 tensor that contains start and end ids of the nodes in the latest layer.

public Output<Integer> numAttemptedLayers ()

The number of layers we attempted to build (but not necessarily succeeded).

public Output<Integer> numFinalizedTrees ()

The number of trees that were finished successfully.

public Output<Integer> numTrees ()

The number of trees in the tree ensemble resource.

public Output<Long> stampToken ()

Stamp token of the tree ensemble resource.