R/asserts.R

In mlr-org/mlrng: Machine Learning in R - Next Generation

#FIXME: we should also do all syntactic sugar for auto-arg conversions here

assertTask = function(task, subclass = "Task") {
  assert_r6(task, subclass)
}


assertLearner = function(learner, subclass = "Learner", for.task = NULL) {
  assert_r6(learner, subclass)
  if (!is.null(for.task)) {
    if (learner$task.type  != for.task$task.type)
      gstop("Types do not match: learner id={learner$id}, task.type={learner$task.type} vs task id={for.task$id}, task.type={for.task$task.type}")
  }
  invisible(learner)
}


assertIndexSet = function(subset, for.task = NULL) {
  if (!is.null(subset)) {
    assertAtomicVector(subset)
  }
  # FIXME: for.task currently ignored
  invisible(subset)
}

assertMeasures = function(measures, for.task = NULL, for.learner = NULL) {
  assertList(measures, "Measure")
  if (!is.null(for.task)) {
    for (m in measures)
    if (for.task$task.type %nin% m$task.types)
      gstop("Types do not match: measure id={m$id}, task.types={stri_flatten(m$task.types, ",")} vs task id={for.task$id}, task.type={for.task$task.type}")
  }
  #FIXME: check that measure matches the learner output
  invisible(measures)
}

assertResampling = function(resampling, for.task = NULL) {
  assertR6(resampling, "Resampling")
}