knn_classifier 0.1.2

This simple library is a classifier for the k-Nearest Neighbors (kNN/k-nn) algorithm.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

use knn_classifier::KnnClassifier;
fn main() {
    // Create the classifier
    let mut clf = KnnClassifier::new(3);
    // Learn from data
    clf.fit(
        &[&[170., 60.], &[166., 58.], &[152., 99.], &[163., 95.], &[150., 90.]],
        &["Normal", "Normal", "Obesity", "Obesity", "Obesity"]);
    // Predict
    let labels = clf.predict(&[vec![159., 85.], vec![165., 55.]]);
    println!("{:?}", labels); // ["Obesity", "Normal"]
    assert_eq!(labels, ["Obesity", "Normal"]);

    // Convert Data to CSV
    let s = clf.to_csv(',');
    println!("{}", s);

    // Convert from CSV
    clf.from_csv(&s, ',', 0, false);
    // Predict one
    let label = clf.predict_one(&[150., 80.]);
    assert_eq!(label, "Obesity");
}