Struct Logistic

pub struct Logistic {
    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 Logistic

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

Create new instance of logistic regression

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

Predict output for logistic regression

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

Save model parameters for 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<Logistic>

Load model parameters for logistic regression

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

Train model parameters for logistic regression

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

Train model parameters for logistic regression with batch gradient descent