xgboost-rust 0.1.0

Rust bindings for XGBoost, a gradient boosting library for machine learning. Downloads XGBoost binaries at build time for cross-platform compatibility.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

// Include the XGBoost C API bindings
mod sys;

mod error;
pub use crate::error::{XGBoostError, XGBoostResult};

mod model;
pub use crate::model::Booster;

// Re-export prediction option constants for convenience
pub mod predict_option {
    /// Normal prediction, output is the transformed probability
    pub const OUTPUT_MARGIN: u32 = 0x01;
    /// Output the untransformed margin value
    pub const PRED_LEAF: u32 = 0x02;
    /// Output the leaf index of trees
    pub const PRED_CONTRIBS: u32 = 0x04;
    /// Output feature contributions (SHAP values)
    pub const PRED_APPROX_CONTRIBS: u32 = 0x08;
    /// Output feature interaction contributions
    pub const PRED_INTERACTIONS: u32 = 0x10;
}