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
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

use treeboost::defaults::linear as linear_defaults;
use treeboost::{LinearConfig, LinearPreset};

#[test]
fn ridge_preset_sets_l1_ratio() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::Ridge);
    assert_eq!(cfg.l1_ratio, linear_defaults::DEFAULT_L1_RATIO);
}

#[test]
fn lasso_preset_sets_l1_ratio() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::Lasso);
    assert_eq!(cfg.l1_ratio, linear_defaults::LASSO_L1_RATIO);
}

#[test]
fn elastic_net_preset_sets_l1_ratio() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::ElasticNet);
    assert_eq!(cfg.l1_ratio, linear_defaults::ELASTIC_NET_L1_RATIO);
}

#[test]
fn aggressive_preset_sets_shrinkage() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::Aggressive);
    assert_eq!(cfg.shrinkage_factor, linear_defaults::AGGRESSIVE_SHRINKAGE);
}

#[test]
fn conservative_preset_sets_shrinkage() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::Conservative);
    assert_eq!(cfg.shrinkage_factor, linear_defaults::CONSERVATIVE_SHRINKAGE);
}

#[test]
fn safe_ridge_sets_damping() {
    let cfg = LinearConfig::default().with_preset(LinearPreset::SafeRidge);
    assert_eq!(
        cfg.extrapolation_damping,
        linear_defaults::SAFE_EXTRAPOLATION_DAMPING
    );
}