treeboost 0.1.0

High-performance Gradient Boosted Decision Tree engine for large-scale tabular data
Documentation 
1
2
3
4
5
6
7
8
9

// Tree overrides for linear trees
pub const LINEAR_TREE_MAX_DEPTH: usize = 4;
pub const LINEAR_TREE_MAX_LEAVES: usize = 15;
pub const LINEAR_TREE_MIN_SAMPLES_LEAF: usize = 20;

// Linear leaf fitting
pub const LINEAR_TREE_LAMBDA: f32 = 1.0;
pub const LINEAR_TREE_MAX_ITER: usize = 50;
pub const MIN_SAMPLES_FOR_LINEAR: usize = 10;