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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
//! A small, pure-Rust gradient boosting library with a deliberately narrow
//! scope: **GBDT only, binary classification only, CPU only, dense numerical
//! features**. No DART/GOSS/RF, no multiclass, no ranking, no regression, no
//! sparse inputs, no GPU, no FFI bindings.
//!
//! # Quickstart
//!
//! Declare your features as named closures over your own row type, then train
//! and predict by handing the same [`FeatureBuilder`] to both sides.
//!
//! ```no_run
//! use nanogbm::{Config, FeatureBuilder, GbdtTrainer};
//!
//! struct Row { age: f64, income: f64, active: bool }
//! # let rows: Vec<Row> = Vec::new();
//! # let labels: Vec<f32> = Vec::new();
//!
//! let cfg = Config {
//! num_iterations: 100,
//! learning_rate: 0.1,
//! num_leaves: 31,
//! ..Config::default()
//! };
//!
//! let fb = FeatureBuilder::<Row>::new()
//! .add_continuous("age", |r| r.age)
//! .add_continuous("income", |r| r.income)
//! .add_boolean("active", |r| r.active);
//!
//! let model = GbdtTrainer::new(&cfg, &fb).fit(&rows, &labels, None)?;
//! let probs = model.predict_proba(&fb, &rows);
//! # Ok::<(), nanogbm::Error>(())
//! ```
//!
//! # Highlights
//!
//! - Histogram learner with sibling-by-subtraction.
//! - Missing values handled at the split (NaN bucket, per-node direction by gain).
//! - Early stopping truncates the model to the best iteration.
//! - Deterministic: same [`Config`] + same data → byte-identical model. All
//! randomness flows through a single `ChaCha8Rng` seeded from
//! [`Config::seed`].
//! - Bincode v2 + serde serialization of [`Model`].
//!
//! See the `examples/` directory for runnable end-to-end programs.
pub use GbdtTrainer;
pub use Config;
pub use Dataset;
pub use ;
pub use FeatureBuilder;
pub use Model;