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
12
13
14
15
16
// Standard
pub const DEFAULT_MAX_SAMPLE_ROWS: usize = 20_000;
pub const DEFAULT_LINEAR_MAX_ITER: usize = 200;
pub const DEFAULT_TREE_MAX_DEPTH: usize = 6;
pub const DEFAULT_TOP_FEATURES: usize = 50;
pub const DEFAULT_ANALYSIS_SEED: u64 = 42;

// Fast/Thorough presets
pub const FAST_MAX_SAMPLE_ROWS: usize = 5_000;
pub const FAST_LINEAR_MAX_ITER: usize = 20;
pub const FAST_TREE_MAX_DEPTH: usize = 3;
pub const FAST_TOP_FEATURES: usize = 20;
pub const THOROUGH_MAX_SAMPLE_ROWS: usize = 50_000;
pub const THOROUGH_LINEAR_MAX_ITER: usize = 100;
pub const THOROUGH_TREE_MAX_DEPTH: usize = 5;
pub const THOROUGH_TOP_FEATURES: usize = 100;