cuda_ml_nvforest_predict_per_tree: Return individual-tree predictions

In cuda.ml: R Interface for the RAPIDS cuML Suite of Libraries

Return individual-tree predictions

Description

Return individual-tree predictions

Usage

cuda_ml_nvforest_predict_per_tree(object, new_data, chunk_size = NULL)

Arguments

object	An nvForest-backed model.
new_data	Numeric predictor data.
chunk_size	Native prediction chunk size, or NULL for the model default. It controls native batching and does not limit the size of the returned R object.

Value

For scalar-leaf models, a numeric matrix with one column per tree. For vector-leaf models, a numeric array indexed by observation, tree, and output.

Memory use

The complete result is materialized in R: rows by trees for scalar-leaf models and rows by trees by outputs for vector-leaf models. The chunk_size argument controls native prediction work but does not bound the memory required by the R result.

See Also

cuda_ml_nvforest_leaf_ids()

cuda.ml documentation built on Aug. 21, 2026, 9:14 a.m.