R/make-model-matrices.R

In tqchen07/bis557: Package Encapsulating work for BIS557

#' @title Make Model Matrices
#' @description Make design matrices and outcome for regression models
#' @param form a formula object, e.g. y ~ x1 + x2
#' @param data a data frame
#'
#' @examples
#' data(iris)
#' make_model_matrices(form = Sepal.Length ~ ., data = iris)
#' @export

make_model_matrices <- function(form, data){
  d_no_na <- model.frame(form, data)
  X <- model.matrix(form, d_no_na)
  y_name <- as.character(form)[2]
  Y <- matrix(d_no_na[, y_name], ncol = 1)
  list(X = X, Y = Y)
}