Struct SGDRegressor 

pub struct SGDRegressor {
    pub loss: SGDRegressorLoss,
    pub penalty: Penalty,
    pub alpha: f32,
    pub l1_ratio: f32,
    pub max_iter: usize,
    pub tol: f32,
    pub learning_rate: LearningRate,
    pub eta0: f32,
    pub power_t: f32,
    pub epsilon: f32,
    pub shuffle: bool,
    /* private fields */
}

SGD Regressor - Linear regressor with Stochastic Gradient Descent

Fields

loss: SGDRegressorLoss

penalty: Penalty

alpha: f32

l1_ratio: f32

max_iter: usize

tol: f32

learning_rate: LearningRate

eta0: f32

power_t: f32

epsilon: f32

shuffle: bool

Implementations

impl SGDRegressor

pub fn new() -> Self

pub fn loss(self, l: SGDRegressorLoss) -> Self

pub fn penalty(self, p: Penalty) -> Self

pub fn alpha(self, a: f32) -> Self

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

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

Trait Implementations

impl Default for SGDRegressor

fn default() -> Self

Returns the “default value” for a type.