Skip to main content

Crate forestfire_core

Crate forestfire_core 

Source
Expand description

ForestFire core model, training, inference, and interchange layer.

The crate is organized around a few stable abstractions:

  • forestfire_data::TableAccess is the common data boundary for both training and inference.
  • TrainConfig is the normalized configuration surface shared by the Rust and Python APIs.
  • Model is the semantic model view used for exact prediction, serialization, and introspection.
  • OptimizedModel is 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§

GradientBoostedTrees
Stage-wise gradient-boosted tree ensemble.
NumericBinBoundary
OptimizedModel
Runtime-lowered model used for faster inference.
PredictionHistogramEntry
PredictionValueStats
RandomForest
Bagged ensemble of decision trees.
TrainConfig
Unified training configuration shared by the Rust and Python entry points.
TreeStructureSummary

Enums§

BoostingError
CompiledArtifactError
Criterion
FeaturePreprocessing
InputFeatureKind
IntrospectionError
MaxFeatures
Model
Top-level semantic model enum.
OptimizeError
PredictError
Task
TrainAlgorithm
TrainError
TreeType

Functions§

train