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
10
11

pub const DEFAULT_MAX_NEW_FEATURES: usize = 50;
pub const LOW_LINEAR_R2_THRESHOLD: f32 = 0.3;
pub const RATIO_CORRELATION_THRESHOLD: f32 = 0.5;
pub const TOP_N_POLYNOMIAL: usize = 5;
pub const TOP_N_INTERACTIONS: usize = 10;
pub const MINIMAL_MAX_NEW_FEATURES: usize = 20;
pub const AGGRESSIVE_MAX_NEW_FEATURES: usize = 100;
pub const AGGRESSIVE_LOW_LINEAR_R2_THRESHOLD: f32 = 0.2;
pub const AGGRESSIVE_RATIO_CORRELATION_THRESHOLD: f32 = 0.4;
pub const AGGRESSIVE_TOP_N_POLYNOMIAL: usize = 10;
pub const AGGRESSIVE_TOP_N_INTERACTIONS: usize = 20;