public final class
BoostedTreesUpdateEnsemble
Updates the tree ensemble by either adding a layer to the last tree being grown
or by starting a new tree.
Public Methods
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 wrapping a new BoostedTreesUpdateEnsemble operation.
|
Inherited Methods
Public Methods
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 wrapping a new BoostedTreesUpdateEnsemble 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. |
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