Struct MultiClassLogistic

pub struct MultiClassLogistic {
    pub features: Value<NDArray<f64>>,
    pub outputs: Value<NDArray<f64>>,
    pub weights: Value<NDArray<f64>>,
    pub bias: Value<NDArray<f64>>,
    /* private fields */
}

Fields

features: Value<NDArray<f64>>

outputs: Value<NDArray<f64>>

weights: Value<NDArray<f64>>

bias: Value<NDArray<f64>>

Implementations

impl MultiClassLogistic

pub fn new( features: &NDArray<f64>, y: &NDArray<f64>, activation_function: fn(values: NDArray<f64>) -> NDArray<f64>, learning_rate: f64, ) -> Result<MultiClassLogistic, String>

Create instance of multi class logistic regression

pub fn predict(&mut self, inputs: NDArray<f64>) -> NDArray<f64>

Make prediction for multi class logistic regression

pub fn save(&self, filepath: &str) -> Result<()>

Save model parameters for multi class logistic regression

pub fn load( filepath: &str, features: NDArray<f64>, y: NDArray<f64>, activation_function: fn(values: NDArray<f64>) -> NDArray<f64>, learning_rate: f64, ) -> Result<MultiClassLogistic>

Load model parameters for multi class logistic regression

pub fn train(&mut self, epochs: usize, log_output: bool)

Train model parameters for multi class logistic regression

pub fn sgd(&mut self, epochs: usize, log_output: bool, batch_size: usize)

Train model paramaters with batch gradient descent for multi class logistic regression