Expand description
ForestFire core model, training, inference, and interchange layer.
The crate is organized around a few stable abstractions:
forestfire_data::TableAccessis the common data boundary for both training and inference.TrainConfigis the normalized configuration surface shared by the Rust and Python APIs.Modelis the semantic model view used for exact prediction, serialization, and introspection.OptimizedModelis a lowered runtime view used when prediction speed matters more than preserving the original tree layout.
Keeping the semantic model and the runtime model separate is deliberate. It makes export and introspection straightforward while still allowing the optimized path to use layouts that are awkward to serialize directly.
Re-exports§
pub use ir::IrError;pub use ir::ModelPackageIr;pub use tree::classifier::DecisionTreeAlgorithm;pub use tree::classifier::DecisionTreeClassifier;pub use tree::classifier::DecisionTreeError;pub use tree::classifier::DecisionTreeOptions;pub use tree::classifier::train_c45;pub use tree::classifier::train_cart;pub use tree::classifier::train_id3;pub use tree::classifier::train_oblivious;pub use tree::classifier::train_randomized;pub use tree::regressor::DecisionTreeRegressor;pub use tree::regressor::RegressionTreeAlgorithm;pub use tree::regressor::RegressionTreeError;pub use tree::regressor::RegressionTreeOptions;pub use tree::regressor::train_cart_regressor;pub use tree::regressor::train_oblivious_regressor;pub use tree::regressor::train_randomized_regressor;
Modules§
- ir
- Stable intermediate representation for ForestFire models.
- tree
- Tree learners used across the project.
Structs§
- Gradient
Boosted Trees - Stage-wise gradient-boosted tree ensemble.
- Numeric
BinBoundary - Optimized
Model - Runtime-lowered model used for faster inference.
- Prediction
Histogram Entry - Prediction
Value Stats - Random
Forest - Bagged ensemble of decision trees.
- Train
Config - Unified training configuration shared by the Rust and Python entry points.
- Tree
Structure Summary
Enums§
- Boosting
Error - Compiled
Artifact Error - Criterion
- Feature
Preprocessing - Input
Feature Kind - Introspection
Error - MaxFeatures
- Missing
Value Strategy - Missing
Value Strategy Config - Model
- Top-level semantic model enum.
- Optimize
Error - Predict
Error - Task
- Train
Algorithm - Train
Error - Tree
Type