todo-files/test_classif_plsDA.R

In guillermozbta/mir: Machine Learning in R

context("classif_plsDA")

test_that("classif_plsDA", {
  requirePackages("DiscriMiner", default.method = "load")
  set.seed(getOption("mlr.debug.seed"))
  m = DiscriMiner::plsDA(multiclass.train[,-multiclass.class.col], group=multiclass.train[,multiclass.class.col])
  p = DiscriMiner::classify(m, newdata=multiclass.test[,-multiclass.class.col])
  testSimple("classif.plsDA", multiclass.df, multiclass.target, multiclass.train.inds, p$pred_class)

  tt = function (formula, data, subset, ...) {
    j = which(colnames(data) == as.character(formula)[2])
    m = DiscriMiner::plsDA(variables = data[subset,-j], group = data[subset,j])
    list(model = m, target=j)
  }

  tp = function(model, newdata) {
    DiscriMiner::classify(model$model, newdata = newdata[,-model$target])$pred_class
  }

  testCV("classif.plsDA", multiclass.df, multiclass.target, tune.train=tt, tune.predict=tp)
})