tests/testthat/test_ridgereg.R

In Jorisvdoorn/bonuslabgroup8: Introduction to Machine Learning in R

context("ridgereg.R")

library(MASS)

test_that("Results are similar to lm.ridge", {
  mod_object = lm.ridge(Petal.Length~Species, data = iris, lambda = 0.1)
  reg1 = ridgereg(Petal.Length~Species, data = iris, lambda = 0.1)
  expect_true(mean(abs(mod_object$coef - reg1$beta_hat)) < 0.01)
  
  mod_object = lm.ridge(Petal.Length~Species, data = iris, lambda = 0.01)
  reg1 = ridgereg(Petal.Length~Species, data = iris, lambda = 0.01)
  expect_true(mean(abs(mod_object$coef - reg1$beta_hat)) < 0.01)
  
  mod_object = lm.ridge(Petal.Length~Species, data = iris, lambda = 0.5)
  reg1 = ridgereg(Petal.Length~Species, data = iris, lambda = 0.5)
  expect_true(mean(abs(mod_object$coef - reg1$beta_hat)) < 0.01)
})