reductionml-core 0.1.0

Reduction based machine learning toolkit core library
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
26
27
28

use crate::{
    types::{Label, Prediction},
    Features,
};

pub trait Metric {
    fn add_point(&mut self, features: &Features, label: &Label, prediction: &Prediction);
    fn get_value(&self) -> MetricValue;
    fn get_name(&self) -> String;
}

pub enum MetricValue {
    Bool(bool),
    Float(f32),
    Int(i32),
    String(String),
}

impl ToString for MetricValue {
    fn to_string(&self) -> String {
        match self {
            MetricValue::Bool(b) => b.to_string(),
            MetricValue::Float(f) => f.to_string(),
            MetricValue::Int(i) => i.to_string(),
            MetricValue::String(s) => s.clone(),
        }
    }
}