rafor 0.3.0

Fast Random Forest library.
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

use rafor::dt::Classifier;
use rafor::prelude::*; // Required for .with_option builders.

fn main() {
    // We have 5 samples with 3 classes.
    #[rustfmt::skip]
    let dataset = [
        0.7, 0.0, 
        0.8, 1.0, 
        0.3, 0.0,
        1.0, 1.3,
        0.4, 2.1
    ];
    let targets = [1, 5, 1, -15, 5];
    let predictor = Classifier::trainer()
        .with_max_depth(15)
        .train(&dataset, &targets);

    // Get predictions for same dataset.
    let predictions = predictor.predict(&dataset);
    println!("Predictions: {:?}", predictions);

    let proba = predictor.proba(&dataset);
    println!("Probability distributions:");
    for p in proba.chunks(predictor.num_classes()) {
        println!("{:?}", p);
    }
}