Nothing
#' @template surv_measure
#' @templateVar title Mean Squared Error
#' @templateVar fullname MeasureSurvMSE
#'
#' @template param_se
#'
#' @description
#' Calculates the mean squared error (MSE).
#'
#' The MSE is defined by
#' \deqn{\frac{1}{n} \sum ((t - \hat{t})^2)}{1/n \sum ((t - t*)^2)}
#' where \eqn{t} is the true value and \eqn{\hat{t}}{t*} is the prediction.
#'
#' Censored observations in the test set are ignored.
#'
#' @family response survival measures
#' @export
MeasureSurvMSE = R6::R6Class("MeasureSurvMSE",
inherit = MeasureSurv,
public = list(
#' @description Creates a new instance of this [R6][R6::R6Class] class.
initialize = function() {
ps = ps(
se = p_lgl(default = FALSE)
)
ps$values$se = FALSE
super$initialize(
id = "surv.mse",
range = c(0, Inf),
minimize = TRUE,
predict_type = "response",
man = "mlr3proba::mlr_measures_surv.mse",
label = "Mean Squared Error",
param_set = ps
)
}
),
private = list(
.score = function(prediction, ...) {
if (self$param_set$values$se) {
surv_mse(prediction$truth, prediction$response)$se
} else {
mean(surv_mse(prediction$truth, prediction$response)$mse)
}
}
)
)
Any scripts or data that you put into this service are public.
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.