linfa-ensemble 0.8.0

A general method for creating ensemble classifiers
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

use linfa::prelude::{Fit, Predict, ToConfusionMatrix};
use linfa_ensemble::EnsembleLearnerParams;
use linfa_trees::DecisionTree;
use ndarray_rand::rand::SeedableRng;
use rand::rngs::SmallRng;

fn main() {
    // Number of models in the ensemble
    let ensemble_size = 100;
    // Proportion of training data given to each model
    let bootstrap_proportion = 0.7;

    // Load dataset
    let mut rng = SmallRng::seed_from_u64(42);
    let (train, test) = linfa_datasets::iris()
        .shuffle(&mut rng)
        .split_with_ratio(0.8);

    // Train ensemble learner model
    let model = EnsembleLearnerParams::new(DecisionTree::params())
        .ensemble_size(ensemble_size)
        .bootstrap_proportion(bootstrap_proportion)
        .fit(&train)
        .unwrap();

    // Return highest ranking predictions
    let final_predictions_ensemble = model.predict(&test);
    println!("Final Predictions: \n{final_predictions_ensemble:?}");

    let cm = final_predictions_ensemble.confusion_matrix(&test).unwrap();

    println!("{cm:?}");
    println!("Test accuracy: {} \n with default Decision Tree params, \n Ensemble Size: {ensemble_size},\n Bootstrap Proportion: {bootstrap_proportion}",
    100.0 * cm.accuracy());
}