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
17
18
19
20
21
22
23

// Rounds
pub const DEFAULT_NUM_ROUNDS: usize = 100;
pub const DEFAULT_NUM_BINS: usize = 255;

// GOSS
pub const DEFAULT_GOSS_TOP_RATE: f32 = 0.2;
pub const DEFAULT_GOSS_OTHER_RATE: f32 = 0.1;

// Subsampling
pub const DEFAULT_SUBSAMPLE: f32 = 1.0;
pub const LARGE_DATA_SUBSAMPLE: f32 = 0.8;

// Early Stopping
pub const DEFAULT_EARLY_STOPPING_ROUNDS: usize = 0;
pub const MIN_EARLY_STOPPING_TREES: usize = 20;

// Conformal
pub const DEFAULT_CALIBRATION_RATIO: f32 = 0.0;
pub const DEFAULT_CONFORMAL_QUANTILE: f32 = 0.9;
pub const CONFORMAL_CALIBRATION_RATIO: f32 = 0.2;

// Validation
pub const DEFAULT_VALIDATION_RATIO: f32 = 0.0;