Expand description
Tree-based algorithms for sklears
This crate provides implementations of tree-based machine learning algorithms including:
- Decision Trees (CART algorithm)
- Random Forest
- Extra Trees
This is a simplified version focusing on core functionality.
Re-exports§
pub use config::ndarray_to_dense_matrix;pub use config::DecisionTreeConfig;pub use config::MaxFeatures;pub use config::MissingValueStrategy;pub use criteria::ConditionalTestType;pub use criteria::FeatureType;pub use criteria::MonotonicConstraint;pub use criteria::SplitCriterion;pub use decision_tree::DecisionTree;pub use decision_tree::DecisionTreeBuilder;pub use decision_tree::DecisionTreeClassifier;pub use decision_tree::DecisionTreeRegressor;pub use decision_tree::TreeValidator;pub use extra_trees_enhanced::BinningStrategy;pub use extra_trees_enhanced::FeatureBinning;pub use extra_trees_enhanced::RandomizationStrategy;pub use extra_trees_enhanced::SparseCompression;pub use extra_trees_enhanced::SparseConfig;pub use extra_trees_enhanced::SparseFeature;pub use isolation_forest::IsolationForest;pub use isolation_forest::IsolationForestConfig;pub use isolation_forest::MaxSamples;pub use isolation_forest::StreamingIsolationForest;pub use model_tree::LeafModelType;pub use model_tree::ModelTree;pub use model_tree::ModelTreeConfig;pub use model_tree::ModelTreeNode;pub use node::CompactTreeNode;pub use node::CustomSplit;pub use node::SurrogateSplit;pub use node::TreeNode;pub use random_forest::RandomForestClassifier;pub use splits::HyperplaneSplit;
Modules§
- builder
- Tree building algorithms and utilities
- config
- Configuration types and enums for decision trees
- criteria
- Split criteria and constraints for decision trees
- decision_
tree - Decision Tree implementation
- extra_
trees_ enhanced - Enhanced Extra Trees with Advanced Randomization Strategies
- isolation_
forest - Isolation Forest for Anomaly Detection
- model_
tree - Model Trees - Decision Trees with Linear Models in Leaves
- node
- Tree node data structures and compact representations
- parallel
- Parallel processing utilities for tree algorithms
- prelude
- Prelude module for convenient imports
- random_
forest - Auto-generated module structure
- splits
- Split implementations for decision trees