Struct 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>, col_index: &[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_probability_change( &self, row: &[f64], contribs: &mut [f64], missing: &f64, current_logodds: f64, ) -> f64

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

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

pub fn predict_contributions_row_mode_difference( &self, row: &[f64], contribs: &mut [f64], 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], 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], 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 predict_leaf_indices( &self, data: &Matrix<'_, f64>, missing: &f64, ) -> Vec<usize>

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

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

pub fn get_average_leaf_weights(&self, i: usize) -> f64

pub fn calculate_importance_weight( &self, stats: &mut HashMap<usize, (f32, usize)>, )

pub fn calculate_importance_gain( &self, stats: &mut HashMap<usize, (f32, usize)>, )

pub fn calculate_importance_cover( &self, stats: &mut HashMap<usize, (f32, usize)>, )

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.