Struct BayesianRidge 

pub struct BayesianRidge {
    pub n_iter: usize,
    pub tol: f32,
    pub alpha_1: f32,
    pub alpha_2: f32,
    pub lambda_1: f32,
    pub lambda_2: f32,
    pub fit_intercept: bool,
    /* private fields */
}

Bayesian Ridge Regression with automatic relevance determination

Fields

n_iter: usize

tol: f32

alpha_1: f32

alpha_2: f32

lambda_1: f32

lambda_2: f32

fit_intercept: bool

Implementations

impl BayesianRidge

pub fn new() -> Self

pub fn n_iter(self, n: usize) -> Self

pub fn fit(&mut self, x: &Tensor, y: &Tensor)

pub fn predict(&self, x: &Tensor) -> Tensor

pub fn predict_with_std(&self, x: &Tensor) -> (Tensor, Tensor)

pub fn score(&self, x: &Tensor, y: &Tensor) -> f32

Trait Implementations

impl Default for BayesianRidge

fn default() -> Self

Returns the “default value” for a type.