mod lab_boosted_classifier;
mod surf_mlp_classifier;
pub use self::lab_boosted_classifier::LabBoostedClassifier;
pub use self::surf_mlp_classifier::SurfMlpClassifier;
pub use self::surf_mlp_classifier::SurfMlpBuffers;
#[derive(Debug, Hash, PartialEq, Eq, Clone)]
pub enum ClassifierKind {
LabBoosted,
SurfMlp,
}
impl ClassifierKind {
#[inline]
pub fn from(id: i32) -> Option<Self> {
match id {
0 => Some(ClassifierKind::LabBoosted),
1 => Some(ClassifierKind::SurfMlp),
_ => None,
}
}
}
pub struct Score {
positive: bool,
score: f32,
}
impl Score {
#[inline]
pub fn is_positive(&self) -> bool {
self.positive
}
#[inline]
pub fn score(&self) -> f32 {
self.score
}
}
#[derive(Clone)]
pub enum Classifier {
SurfMlp(SurfMlpClassifier),
LabBoosted(LabBoostedClassifier),
}