R/regr_msle.R

Defines functions msle

Documented in msle

#' @title Mean Squared Log Error
#'
#' @details
#' The Mean Squared Log Error is defined as \deqn{
#'   \frac{1}{n} \sum_{i=1}^n w_i \left( \ln (1 + t_i) - \ln (1 + r_i) \right)^2.
#' }{
#'   weighted.mean((log(1 + t) - log(1 + r))^2, weights).
#' }
#'
#' @templateVar mid msle
#' @template regr_template
#'
#' @details
#' This measure is undefined if any element of \eqn{t} or \eqn{r} is less than or equal to \eqn{-1}.
#'
#' @inheritParams regr_params
#' @template regr_example
#' @export
msle = function(truth, response, sample_weights = NULL, na_value = NaN, ...) {
  assert_regr(truth, response = response, na_value = na_value)
  if (min(truth, response) <= -1) {
    return(na_value)
  }
  wmean(.sle(truth, response), sample_weights)
}

#' @include measures.R
add_measure(msle, "Mean Squared Log Error", "regr", 0, Inf, TRUE, obs_loss = "sle")

Try the mlr3measures package in your browser

Any scripts or data that you put into this service are public.

mlr3measures documentation built on Aug. 5, 2022, 5:22 p.m.