Function classify

src/knn.rs:61–72

(m: &Matrix<T>, labels: &Vec<L>, example: &[T], k: usize, df: D)

}

pub fn classify<T, L, D>(m: &Matrix<T>, labels: &Vec<L>, example: &[T], k: usize, df: D) -> L
    where T: Float, L: Clone + Ord, D: Fn(&[T], &[T]) -> T {

    let idx = scan(&m, example, k, df).unwrap();

    let mut targets: Vec<L> = idx.iter().map(|pos| labels.get(*pos).unwrap()).cloned().collect();
    targets.sort_by(|a, b| a.cmp(&b));
    let mut r = group(&targets);
    r.sort_by(|a, b| a.1.cmp(&b.1));

    r.last().unwrap().0.clone()
}


#[cfg(test)]