R/ResamplingBase.R

In mlr3resampling: Resampling Algorithms for 'mlr3' Framework

ResamplingBase = R6::R6Class(
  "ResamplingBase",
  inherit = mlr3::Resampling,
  public = list(
    train_set = function(i) {
      self$instance$iteration.dt$train[[i]]
    },
    test_set = function(i) {
      self$instance$iteration.dt$test[[i]]
    }
  ),
  active = list(
    iters = function(rhs) {
      nrow(self$instance$iteration.dt)
    }
  ),
  private = list(
    .sample = function(ids, ...) {
      data.table(
        row_id = ids,
        fold = sample(
          seq(0, length(ids)-1) %%
            as.integer(self$param_set$values$folds) + 1L
        ),
        key = "fold"
      )
    }
  )
)