Struct TreeEnsemble 

pub struct TreeEnsemble { /* private fields */ }

Implementations

impl TreeEnsemble

pub fn new(trees: Vec<(Tree, f64)>, base_values: Vec<f64>) -> Result<Self>

pub fn new_with_output_groups( trees: Vec<(Tree, f64)>, base_values: Vec<f64>, output_groups: Vec<Option<usize>>, ) -> Result<Self>

pub fn trees(&self) -> &[(Tree, f64)]

pub fn output_groups(&self) -> &[Option<usize>]

pub fn base_offset(&self) -> ArrayView1<'_, f64>

pub fn n_features(&self) -> usize

pub fn n_outputs(&self) -> usize

pub fn expected_value(&self) -> Array1<f64>

pub fn predict_with_base_margin( &self, x: ArrayView2<'_, f64>, base_margin: ArrayView2<'_, f64>, ) -> Result<Array2<f64>>

Predicts raw tree outputs with a per-sample base margin replacing the ensemble’s fixed base offset (matching XGBoost base_margin semantics).

pub fn validate(&self) -> Result<()>

Revalidates all trees and ensemble dimensions after deserialization.

Trait Implementations

impl Clone for TreeEnsemble

fn clone(&self) -> TreeEnsemble

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TreeEnsemble

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TreeEnsemble

fn deserialize<D: Deserializer<'de>>(deserializer: D) -> Result<Self, D::Error>

Deserialize this value from the given Serde deserializer.

impl Predict for TreeEnsemble

fn predict(&self, x: ArrayView2<'_, f64>) -> Result<Array2<f64>>

Predict outputs for a batch of samples.

fn n_features(&self) -> Option<usize>

Returns the number of input features expected by the model.

fn n_outputs(&self) -> Option<usize>

Returns the number of outputs produced by the model.

fn predict_owned(&self, x: Array2<f64>) -> Result<Array2<f64>>

Predicts from an owned batch.

impl Serialize for TreeEnsemble

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.