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>, cuts: &JaggedMatrix<f64>, grad: &[f32], hess: &[f32], splitter: &T, max_leaves: usize, max_depth: usize, parallel: bool )

pub fn predict_contributions_row( &self, data: &Matrix<'_, f64>, row: usize, contribs: &mut [f64], weights: &[f64] )

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

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

pub fn value_partial_dependence(&self, feature: usize, value: 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.