Skip to main content

predict_batch

Function predict_batch 

Source
pub fn predict_batch(
    arena: &TreeArena,
    root: NodeId,
    feature_matrix: &[Vec<f64>],
) -> Vec<f64>
Available on crate feature alloc only.
Expand description

Batch prediction: compute the leaf value for each row in feature_matrix.

Returns a Vec<f64> with one prediction per row, in the same order as the input matrix.