regInterceptModel {heuristica}

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.