rustlearn 0.3.0

A machine learning package 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

//! Common rustlearn traits.

use std::cmp::Ordering;

use array::dense::*;


pub trait SupervisedModel<T> {
    fn fit(&mut self, X: &T, y: &Array) -> Result<(), &'static str>;
    fn decision_function(&self, X: &T) -> Result<Array, &'static str>;
    fn predict(&self, x: &T) -> Result<Array, &'static str> {

        let decision_func = try!(self.decision_function(x));

        Ok(Array::from(decision_func.data()
                                    .iter()
                                    .map(|v| {
                                        match v.partial_cmp(&0.5) {
                                            Some(Ordering::Greater) => 1.0,
                                            _ => 0.0,
                                        }
                                    })
                                    .collect::<Vec<f32>>()))
    }
}