linfa-preprocessing 0.8.1

A Machine Learning framework for Rust
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

use linfa::metrics::ToConfusionMatrix;
use linfa::traits::{Fit, Predict, Transformer};
use linfa_bayes::GaussianNb;
use linfa_preprocessing::linear_scaling::LinearScaler;

fn main() {
    // Read in the dataset and convert continuous target into categorical
    let (train, valid) = linfa_datasets::winequality()
        .map_targets(|x| if *x > 6 { 1 } else { 0 })
        .split_with_ratio(0.7);

    // Fit a standard scaler to the training set
    let scaler = LinearScaler::standard().fit(&train).unwrap();

    // Scale training and validation sets according to the fitted scaler
    let train = scaler.transform(train);
    let valid = scaler.transform(valid);

    // Learn a naive bayes model from the training set
    let model = GaussianNb::params().fit(&train).unwrap();

    // compute accuracies
    let train_acc = model
        .predict(&train)
        .confusion_matrix(&train)
        .unwrap()
        .accuracy();
    let cm = model.predict(&valid).confusion_matrix(&valid).unwrap();
    let valid_acc = cm.accuracy();
    println!("Scaled model training and validation accuracies: {train_acc} - {valid_acc}");
    println!("{cm:?}",);
}