Skip to main content

dtrees_rs/algorithms/greedy/lgdt/
factories.rs

1//! Ready-made LGDT builders.
2
3use crate::algorithms::common::errors::NativeError;
4use crate::algorithms::greedy::lgdt::builder::LGDTBuilder;
5use crate::algorithms::optimal::depth2::{ErrorMinimizer, InfoGainMaximizer};
6
7/// An LGDT builder whose lookahead maximises information gain.
8pub fn with_info_gain() -> LGDTBuilder<InfoGainMaximizer<NativeError>> {
9    LGDTBuilder::default().search(Box::<InfoGainMaximizer<NativeError>>::default())
10}
11
12/// An LGDT builder whose lookahead minimises the error.
13pub fn with_error_minimizer() -> LGDTBuilder<ErrorMinimizer<NativeError>> {
14    LGDTBuilder::default().search(Box::<ErrorMinimizer<NativeError>>::default())
15}