Struct forust_ml::tree::Tree

pub struct Tree {
    pub nodes: Vec<Node>,
}

Fields

nodes: Vec<Node>

Implementations

impl Tree

pub fn new() -> Self

pub fn fit<T: Splitter>( &mut self, data: &Matrix<'_, u16>, index: Vec<usize>, cuts: &JaggedMatrix<f64>, grad: &[f32], hess: &[f32], splitter: &T, max_leaves: usize, max_depth: usize, parallel: bool, sample_method: &SampleMethod, grow_policy: &GrowPolicy )

pub fn predict_contributions_row_midpoint_difference( &self, row: &[f64], contribs: &mut [f64], missing: &f64 )

pub fn predict_contributions_midpoint_difference( &self, data: &Matrix<'_, f64>, contribs: &mut [f64], parallel: bool, missing: &f64 )

pub fn predict_contributions_row_branch_difference( &self, row: &[f64], contribs: &mut [f64], missing: &f64 )

pub fn predict_contributions_branch_difference( &self, data: &Matrix<'_, f64>, contribs: &mut [f64], parallel: bool, missing: &f64 )

pub fn predict_contributions_row_weight( &self, row: &[f64], contribs: &mut [f64], missing: &f64 )

pub fn predict_contributions_weight( &self, data: &Matrix<'_, f64>, contribs: &mut [f64], parallel: bool, missing: &f64 )

pub fn predict_contributions_row_average( &self, row: &[f64], contribs: &mut [f64], weights: &[f64], missing: &f64 )

This is the method that XGBoost uses.

pub fn predict_contributions_average( &self, data: &Matrix<'_, f64>, contribs: &mut [f64], weights: &[f64], parallel: bool, missing: &f64 )

pub fn predict_row_from_row_slice(&self, row: &[f64], missing: &f64) -> f64

pub fn predict( &self, data: &Matrix<'_, f64>, parallel: bool, missing: &f64 ) -> Vec<f64>

pub fn value_partial_dependence( &self, feature: usize, value: f64, missing: &f64 ) -> f64

pub fn distribute_leaf_weights(&self) -> Vec<f64>

Trait Implementations

impl Default for Tree

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Tree

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for Tree

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

Formats the value using the given formatter.

impl Serialize for Tree

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

Serialize this value into the given Serde serializer.