lightgbm-rust 0.1.1

Rust bindings for LightGBM, a gradient boosting framework. Downloads LightGBM 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

// Include the LightGBM C API bindings
mod sys;

mod error;
pub use crate::error::{LightGBMError, LightGBMResult};

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

// Re-export prediction type constants for convenience
pub mod predict_type {
    /// Normal prediction
    pub const NORMAL: i32 = 0;
    /// Raw score prediction
    pub const RAW_SCORE: i32 = 1;
    /// Leaf index prediction
    pub const LEAF_INDEX: i32 = 2;
    /// Feature contribution (SHAP values)
    pub const CONTRIB: i32 = 3;
}