R/estimateResidualVariance.R

In guillermozbta/s2: Machine Learning in R

#' Estimate the residual variance.
#'
#' Estimate the residual variance of a regression model on a given task.
#' If a regression learner is provided instead of a model, the model is
#' trained (see \code{\link{train}}) first.
#'
#' @param x [\code{\link{Learner}} or \code{\link{WrappedModel}}]\cr
#'   Learner or wrapped model.
#' @param task [\code{\link{RegrTask}}]\cr
#'   Regression task.
#'   If missing, \code{data} and \code{target} must be supplied.
#' @param data [\code{data.frame}]\cr
#'   A data frame containing the features and target variable.
#'   If missing, \code{task} must be supplied.
#' @param target [\code{character(1)}]\cr
#'   Name of the target variable.
#'   If missing, \code{task} must be supplied.
#' @export
estimateResidualVariance = function(x, task, data, target) {
  UseMethod("estimateResidualVariance")
}

#' @export
estimateResidualVariance.Learner = function(x, task, data, target) {
  if (missing(task))
    task = makeRegrTask(data = data, target = target)
  estimateResidualVariance.WrappedModel(train(x, task), task)
}

#' @export
estimateResidualVariance.WrappedModel = function(x, task, data, target) {
  if (missing(task)) {
    task = makeRegrTask(data = data, target = target)
  } else {
    assertClass(task, classes = "RegrTask")
  }
  p = predict(x, task)
  var(p$data$response - p$data$truth)
}