Struct SGDClassifier 

pub struct SGDClassifier {
    pub loss: SGDLoss,
    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 shuffle: bool,
    pub warm_start: bool,
    pub class_weight: Option<Vec<f32>>,
    /* private fields */
}

SGD Classifier - Linear classifier with Stochastic Gradient Descent

Fields

loss: SGDLoss

penalty: Penalty

alpha: f32

l1_ratio: f32

max_iter: usize

tol: f32

learning_rate: LearningRate

eta0: f32

power_t: f32

shuffle: bool

warm_start: bool

class_weight: Option<Vec<f32>>

Implementations

impl SGDClassifier

pub fn new() -> Self

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

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

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

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

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

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

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

Trait Implementations

impl Default for SGDClassifier

fn default() -> Self

Returns the “default value” for a type.