Struct DecisionTreeRegressor 

pub struct DecisionTreeRegressor {
    pub max_depth: Option<usize>,
    pub min_samples_split: usize,
    pub min_samples_leaf: usize,
    pub max_features: Option<usize>,
    pub criterion: Criterion,
    /* private fields */
}

Decision Tree Regressor using CART algorithm

Fields

max_depth: Option<usize>

min_samples_split: usize

min_samples_leaf: usize

max_features: Option<usize>

criterion: Criterion

Implementations

impl DecisionTreeRegressor

pub fn new() -> Self

pub fn max_depth(self, depth: usize) -> Self

pub fn fit(&mut self, x: &Tensor, y: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

Trait Implementations

impl Default for DecisionTreeRegressor

fn default() -> Self

Returns the “default value” for a type.