Struct ErrorMinimizer 

pub struct ErrorMinimizer<E>
where
    E: ErrorWrapper + ?Sized,
{ /* private fields */ }

Depth-2 solver that minimises the error of the tree: the optimal tree of depth at most 2.

Implementations

impl<E> ErrorMinimizer<E>

where E: ErrorWrapper + ?Sized,

pub fn new(error_function: Box<E>) -> Self

A solver using error_function for the error of a leaf.

Trait Implementations

impl Default for ErrorMinimizer<NativeError>

fn default() -> Self

Returns the “default value” for a type.

impl<E> OptimalDepth2Tree for ErrorMinimizer<E>

where E: ErrorWrapper + ?Sized,

fn find_optimal_depth_one_tree( &self, min_sup: usize, cover: &mut Cover, provided_candidates: Option<&[usize]>, ) -> Result<Tree, FitError>

The best tree with a single test.

fn find_optimal_depth_two_tree( &self, min_sup: usize, cover: &mut Cover, provided_candidates: Option<&[usize]>, ) -> Result<Tree, FitError>

The best tree with at most two levels of tests.

fn error(&self, distribution: &[usize]) -> (f64, f64)

(error, prediction) of a leaf with the given class counts.

fn fit( &self, min_sup: usize, depth: usize, cover: &mut Cover, provided_candidates: Option<&[usize]>, ) -> Result<Tree, FitError>

Finds the best tree of the given depth (1 or 2) on the instances of cover, using only provided_candidates when given.

fn get_candidates( &self, cover: &mut Cover, min_sup: usize, provided_candidates: Option<&[usize]>, ) -> Vec<usize>

Features that split the instances of cover with at least min_sup instances on each side.