rs-ml 0.4.0

Simple ML crate including Gaussian Naive Bayesian classifier
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
29

//! Functionality to transform and scale data.

use crate::Estimator;

pub mod embedding;
pub mod scalers;

/// Generic trait to transform data.
pub trait Transformer<Input, Output> {
    /// Transform input data based on previously fitted data
    fn transform(&self, input: &Input) -> Option<Output>;
}

/// Provided trait on estimators that emit a transfomer
pub trait FitTransform<Input, Output, T: Transformer<Input, Output>>:
    Estimator<Input, Estimator = T>
{
    /// Fit and transform data in one operation. Useful if you don't need to use the fitted transformer
    /// multiple times.
    fn fit_transform(&self, input: &Input) -> Option<Output> {
        let transfomer = self.fit(input)?;
        transfomer.transform(input)
    }
}

impl<Input, Output, T: Transformer<Input, Output>, E: Estimator<Input, Estimator = T>>
    FitTransform<Input, Output, T> for E
{
}