rusty_science 0.1.1

An easy to learn and use ML toolkit for rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16

use rusty_science::classification::TreeClassifier;
use rusty_science::data::datasets::load_iris;

#[test]
pub(crate) fn test_tree_classification_integration() {
    let iris_data = load_iris();
    let (data, labels) = iris_data.to_numerical_labels();

    let target = vec![1.5, 1.5, 1.5, 1.5];

    let mut tree = TreeClassifier::new();
    tree.fit(data, labels);
    let prediction = tree.predict(target);

    assert_eq!(prediction, 0);
}