man-roxygen/section_predict_sets.R

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

#' @section Predict Sets:
#' If you want to compare the performance of a learner on the training with the performance
#' on the test set, you have to configure the [Learner] to predict on multiple sets by
#' setting the field `predict_sets` to `c("train", "test")` (default is `"test"`).
#' Each set yields a separate [Prediction] object during resampling.
#' In the next step, you have to configure the measures to operate on the respective Prediction object:
#' ```
#' m1 = msr("classif.ce", id = "ce.train", predict_sets = "train")
#' m2 = msr("classif.ce", id = "ce.test", predict_sets = "test")
#' ```
#'
#' The (list of) created measures can finally be passed to `$aggregate()` or `$score()`.