Struct Linear

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

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

Create new instance of linear regression

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

Predict outcomes for linear regression model

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

Save model parameters for linear regression

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

Load model parameters for linear regression

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

Train model parameters for linear regression

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

Train model parameters for linear regression with batch gradient descent