regInterceptModel: Linear regression wrapper for hueristica

In jeanimal/heuristica: Heuristics Including Take the Best and Unit-Weight Linear

Linear regression wrapper for hueristica

Description

A wrapper to create a lm model just specifying columns, generating a model formula for you. This makes it easier to run automated comparisons with other models in heuristica.

Usage

regInterceptModel(train_matrix, criterion_col, cols_to_fit)

Arguments

train_matrix	A matrix (or data.frame) of data to train (fit) the model with.
criterion_col	The index of the criterion column– "y" in the formula.
cols_to_fit	A vector of column indexes to fit– the "x's" in the formula.

Details

This version assumes you always want to include the intercept.

Value

An object of class regInterceptModel, which is a subclass of lm.

See Also

regModel for a version that excludes the intercept.

predict.lm for prediction.

predictPairProb for predicting between a pair of rows.

jeanimal/heuristica documentation built on Feb. 3, 2024, 9:56 p.m.