sklears-multioutput 0.2.0

Multi-output regression and classification
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

//! # BayesianMultiOutputModel - Trait Implementations
//!
//! This module contains trait implementations for `BayesianMultiOutputModel`.
//!
//! ## Implemented Traits
//!
//! - `Default`
//! - `Estimator`
//! - `Fit`
//! - `Predict`
//! - `Estimator`
//!
//! 🤖 Generated with [SplitRS](https://github.com/cool-japan/splitrs)
#![allow(non_snake_case)] // Standard ML notation: X for feature matrices, K for kernels

use super::*;

impl Default for BayesianMultiOutputModel<Untrained> {
    fn default() -> Self {
        Self::new()
    }
}

impl Estimator for BayesianMultiOutputModel<Untrained> {
    type Config = BayesianMultiOutputConfig;
    type Error = SklearsError;
    type Float = Float;
    fn config(&self) -> &Self::Config {
        &self.config
    }
}

impl Fit<ArrayView2<'_, Float>, ArrayView2<'_, Float>> for BayesianMultiOutputModel<Untrained> {
    type Fitted = BayesianMultiOutputModel<BayesianMultiOutputModelTrained>;
    fn fit(self, X: &ArrayView2<'_, Float>, y: &ArrayView2<'_, Float>) -> SklResult<Self::Fitted> {
        let (n_samples, n_features) = X.dim();
        let (y_samples, n_outputs) = y.dim();
        if n_samples != y_samples {
            return Err(SklearsError::InvalidInput(
                "X and y must have the same number of samples".to_string(),
            ));
        }
        let mut rng = thread_rng();
        let (
            weight_posterior,
            bias_posterior,
            noise_posterior,
            log_marginal_likelihood,
            elbo_history,
        ) = match self.config.inference_method {
            InferenceMethod::Variational => self.variational_inference(X, y, &mut rng)?,
            InferenceMethod::MCMC => self.mcmc_inference(X, y, &mut rng)?,
            InferenceMethod::EM => self.em_inference(X, y, &mut rng)?,
            InferenceMethod::Laplace => self.laplace_inference(X, y, &mut rng)?,
            InferenceMethod::Exact => self.exact_inference(X, y, &mut rng)?,
        };
        Ok(BayesianMultiOutputModel {
            state: BayesianMultiOutputModelTrained {
                weight_posterior,
                bias_posterior,
                noise_posterior,
                n_features,
                n_outputs,
                log_marginal_likelihood,
                elbo_history,
                config: self.config.clone(),
            },
            config: self.config,
        })
    }
}

impl Predict<ArrayView2<'_, Float>, Array2<Float>>
    for BayesianMultiOutputModel<BayesianMultiOutputModelTrained>
{
    fn predict(&self, X: &ArrayView2<'_, Float>) -> SklResult<Array2<Float>> {
        let (_n_samples, n_features) = X.dim();
        if n_features != self.state.n_features {
            return Err(SklearsError::InvalidInput(format!(
                "Expected {} features, got {}",
                self.state.n_features, n_features
            )));
        }
        let bias_1d = self.state.bias_posterior.mean.slice(s![.., 0]);
        let predictions = X.dot(&self.state.weight_posterior.mean) + bias_1d;
        Ok(predictions)
    }
}

impl Estimator for BayesianMultiOutputModel<BayesianMultiOutputModelTrained> {
    type Config = BayesianMultiOutputConfig;
    type Error = SklearsError;
    type Float = Float;
    fn config(&self) -> &Self::Config {
        &self.state.config
    }
}