Struct tract_onnx_opl::ml::tree::TreeEnsemble

pub struct TreeEnsemble {
    pub data: TreeEnsembleData,
    pub max_used_feature: usize,
    pub n_classes: usize,
    pub aggregate_fn: Aggregate,
}

Fields

data: TreeEnsembleData

max_used_feature: usize

n_classes: usize

aggregate_fn: Aggregate

Implementations

impl TreeEnsemble

pub fn build(
    data: TreeEnsembleData,
    max_used_feature: usize,
    n_classes: usize,
    aggregate_fn: Aggregate
) -> TractResult<Self>

pub fn n_classes(&self) -> usize

pub fn check_n_features(&self, n_features: usize) -> TractResult<()>

pub fn eval<'i, I, T>(&self, input: I) -> TractResult<ArrayD<f32>> where
    I: Into<ArrayViewD<'i, T>>,
    T: AsPrimitive<f32>, 

Trait Implementations

impl Clone for TreeEnsemble

fn clone(&self) -> TreeEnsemble

Returns a copy 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 Hash for TreeEnsemble

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H) where
    H: Hasher, 

Feeds a slice of this type into the given Hasher.