scry-learn 0.1.0

Machine learning toolkit in pure Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

// SPDX-License-Identifier: MIT OR Apache-2.0
//! Tree-based models: Decision Tree, Random Forest, Gradient Boosting, and
//! Histogram-based Gradient Boosting.

pub(crate) mod binning;
mod cart;
mod gradient_boosting;
mod histogram_gbt;
mod random_forest;

pub use binning::FeatureBinner;
pub(crate) use cart::FlatTree;
pub use cart::{DecisionTreeClassifier, DecisionTreeRegressor, SplitCriterion, TreeNode};
pub use gradient_boosting::{
    GradientBoostingClassifier, GradientBoostingRegressor, RegressionLoss,
};
pub use histogram_gbt::{
    HistGradientBoostingClassifier, HistGradientBoostingRegressor, HistNodeView,
};
pub use random_forest::{MaxFeatures, RandomForestClassifier, RandomForestRegressor};