Struct LightGBMClassifier 

pub struct LightGBMClassifier {
    pub n_estimators: usize,
    pub learning_rate: f32,
    pub num_leaves: usize,
    pub max_depth: i32,
    pub min_data_in_leaf: usize,
    pub min_sum_hessian_in_leaf: f32,
    pub feature_fraction: f32,
    pub bagging_fraction: f32,
    pub bagging_freq: usize,
    pub lambda_l1: f32,
    pub lambda_l2: f32,
    pub max_bin: usize,
    /* private fields */
}

LightGBM-style Gradient Boosting Classifier

Fields

n_estimators: usize

learning_rate: f32

num_leaves: usize

max_depth: i32

min_data_in_leaf: usize

min_sum_hessian_in_leaf: f32

feature_fraction: f32

bagging_fraction: f32

bagging_freq: usize

lambda_l1: f32

lambda_l2: f32

max_bin: usize

Implementations

impl LightGBMClassifier

pub fn new(n_estimators: usize) -> Self

pub fn learning_rate(self, lr: f32) -> Self

pub fn num_leaves(self, leaves: usize) -> Self

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

pub fn feature_fraction(self, fraction: f32) -> Self

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

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

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