BoostedTreesTrainingPredict

public final class BoostedTreesTrainingPredict

Runs multiple additive regression ensemble predictors on input instances and

computes the update to cached logits. It is designed to be used during training. It traverses the trees starting from cached tree id and cached node id and calculates the updates to be pushed to the cache.

Constants

String OP_NAME The name of this op, as known by TensorFlow core engine

Public Methods

static BoostedTreesTrainingPredict	create(Scope scope, Operand<?> treeEnsembleHandle, Operand<TInt32> cachedTreeIds, Operand<TInt32> cachedNodeIds, Iterable<Operand<TInt32>> bucketizedFeatures, Long logitsDimension) Factory method to create a class wrapping a new BoostedTreesTrainingPredict operation.
Output<TInt32>	nodeIds() Rank 1 Tensor containing new node ids in the new tree_ids.
Output<TFloat32>	partialLogits() Rank 2 Tensor containing logits update (with respect to cached values stored) for each example.
Output<TInt32>	treeIds() Rank 1 Tensor containing new tree ids for each example.

Inherited Methods

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single `Operation`.
final String	toString()

From class java.lang.Object

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()

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single `Operation`.

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "BoostedTreesTrainingPredict"

Public Methods

public static BoostedTreesTrainingPredict create (Scope scope, Operand<?> treeEnsembleHandle, Operand<TInt32> cachedTreeIds, Operand<TInt32> cachedNodeIds, Iterable<Operand<TInt32>> bucketizedFeatures, Long logitsDimension)

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

Parameters

scope	current scope
cachedTreeIds	Rank 1 Tensor containing cached tree ids which is the starting tree of prediction.
cachedNodeIds	Rank 1 Tensor containing cached node id which is the starting node of prediction.
bucketizedFeatures	A list of rank 1 Tensors containing bucket id for each feature.
logitsDimension	scalar, dimension of the logits, to be used for partial logits shape.

Returns

a new instance of BoostedTreesTrainingPredict

public Output<TInt32> nodeIds ()

Rank 1 Tensor containing new node ids in the new tree_ids.

public Output<TFloat32> partialLogits ()

Rank 2 Tensor containing logits update (with respect to cached values stored) for each example.

public Output<TInt32> treeIds ()

Rank 1 Tensor containing new tree ids for each example.