Struct LogisticRegression 

pub struct LogisticRegression {
    pub coef_: Option<Vec<Vec<f32>>>,
    pub intercept_: Option<Vec<f32>>,
    pub penalty: Penalty,
    pub c: f32,
    pub fit_intercept: bool,
    pub max_iter: usize,
    pub tol: f32,
    pub multi_class: MultiClass,
    /* private fields */
}

Logistic Regression for binary and multiclass classification

Fields

coef_: Option<Vec<Vec<f32>>>

intercept_: Option<Vec<f32>>

penalty: Penalty

c: f32

fit_intercept: bool

max_iter: usize

tol: f32

multi_class: MultiClass

Implementations

impl LogisticRegression

pub fn new() -> Self

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

pub fn c(self, c: f32) -> Self

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

pub fn multi_class(self, mc: MultiClass) -> Self

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

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

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

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

Trait Implementations

impl Default for LogisticRegression

fn default() -> Self

Returns the “default value” for a type.