Struct linregress::FormulaRegressionBuilder

pub struct FormulaRegressionBuilder<'a> { /* fields omitted */ }

A builder to create and fit a linear regression model.

Given a dataset and a regression formula this builder will produce an ordinary least squared linear regression model.

See formula and data for details on how to configure this builder.

The pseudo inverse method is used to fit the model.

Usage

use linregress::{FormulaRegressionBuilder, RegressionDataBuilder};

let y = vec![1.,2. ,3. , 4.];
let x = vec![4., 3., 2., 1.];
let data = vec![("Y", y), ("X", x)];
let data = RegressionDataBuilder::new().build_from(data)?;
let model = FormulaRegressionBuilder::new().data(&data).formula("Y ~ X").fit()?;
assert_eq!(model.parameters.intercept_value, 5.0);
assert_eq!(model.parameters.regressor_values[0], -0.9999999999999993);
assert_eq!(model.parameters.regressor_names[0], "X");

Methods

impl<'a> FormulaRegressionBuilder<'a>

pub fn new() -> Self

Create as new FormulaRegressionBuilder with no data or formula set.

pub fn data(self, data: &'a RegressionData<'a>) -> Self

Set the data to be used for the regression.

The data has to be given as a reference to a RegressionData struct. See RegressionDataBuilder for details.

pub fn formula<T: Into<Cow<'a, str>>>(self, formula: T) -> Self

Set the formula to use for the regression.

The expected format is " ~ <regressor 1> + <regressor 2>".

E.g. for a regressand named Y and three regressors named A, B and C the correct format would be "Y ~ A + B + C".

Note that there is currently no special support for categorical variables. So if you have a categorical variable with more than two distinct values or values that are not 0 and 1 you will need to perform "dummy coding" yourself.

pub fn fit(self) -> Result<RegressionModel, Error>

Fits the model and returns a RegressionModel if successful. You need to set the data with data and a formula with formula before you can use it.

Trait Implementations

impl<'a> Default for FormulaRegressionBuilder<'a>

fn default() -> Self

Returns the "default value" for a type.

impl<'a> Clone for FormulaRegressionBuilder<'a>

fn clone(&self) -> FormulaRegressionBuilder<'a>

Returns a copy of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<'a> Debug for FormulaRegressionBuilder<'a>

fn fmt(&self, f: &mut Formatter) -> Result

Formats the value using the given formatter.