rustlearn 0.1.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

//! 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>>()))
    }
}