Skip to main content

scry_learn/tree/
mod.rs

1// SPDX-License-Identifier: MIT OR Apache-2.0
2//! Tree-based models: Decision Tree, Random Forest, Gradient Boosting, and
3//! Histogram-based Gradient Boosting.
4
5pub(crate) mod binning;
6mod cart;
7mod gradient_boosting;
8mod histogram_gbt;
9mod random_forest;
10
11pub use binning::FeatureBinner;
12pub(crate) use cart::FlatTree;
13pub use cart::{DecisionTreeClassifier, DecisionTreeRegressor, SplitCriterion, TreeNode};
14pub use gradient_boosting::{
15    GradientBoostingClassifier, GradientBoostingRegressor, RegressionLoss,
16};
17pub use histogram_gbt::{
18    HistGradientBoostingClassifier, HistGradientBoostingRegressor, HistNodeView,
19};
20pub use random_forest::{MaxFeatures, RandomForestClassifier, RandomForestRegressor};