Struct Ridge

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

pub fn new( features: &NDArray<f64>, y: &NDArray<f64>, lambda: f64, learning_rate: f64, ) -> Result<Ridge, String>

Create new instance of ridge regression

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

Generate prediction for ridge regression

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

Save model parameters for ridge regression

pub fn load( filepath: &str, features: &NDArray<f64>, y: &NDArray<f64>, learning_rate: f64, lambda: f64, ) -> Result<Ridge>

Load model parameters for ridge regression

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

Train model parameters for ridge regression

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

Train model parameters for ridge regression with batch gradient descent