diff --git a/R/ResamplingSptCVCstf.R b/R/ResamplingSptCVCstf.R index 69ba9e432..97f32d3cb 100644 --- a/R/ResamplingSptCVCstf.R +++ b/R/ResamplingSptCVCstf.R @@ -104,6 +104,17 @@ ResamplingSptCVCstf = R6Class("ResamplingSptCVCstf", ) sample_cast = function(task, stratify = FALSE, folds) { + + if (length(task$col_roles$time) && length(task$col_roles$space)) { + lg$info(sprintf("Using column roles 'space' (var. '%s') and 'time' (var. '%s') for partitioning", task$col_roles$space, task$col_roles$time)) + + } else if (length(task$col_roles$time)) { + lg$info(sprintf("Using column role 'time' (var. '%s') for partitioning", task$col_roles$time)) + } + else if (length(task$col_roles$space)) { + lg$info(sprintf("Using column role 'space' (var. '%s') for partitioning", task$col_roles$space)) + } + target = if (stratify) task$target_names else NULL space = task$col_roles$space time = task$col_roles$time diff --git a/R/autoplot_spcv_cstf.R b/R/autoplot_spcv_cstf.R index 9b3a1f41c..f849e5500 100644 --- a/R/autoplot_spcv_cstf.R +++ b/R/autoplot_spcv_cstf.R @@ -118,6 +118,7 @@ autoplot.ResamplingSptCVCstf = function( # nolint resampling_sub$instance = resampling_sub$instance[[repeats_id]] } + # FIXME: check space and time var from column roles! # check if we are in a 2D or 3D scenario if (is.null(plot3D)) { if (!is.null(resampling_sub$space_var) && diff --git a/R/zzz.R b/R/zzz.R index f848f5875..148e06eaa 100644 --- a/R/zzz.R +++ b/R/zzz.R @@ -102,6 +102,7 @@ register_mlr3 = function() { # nocov start .onLoad = function(libname, pkgname) { # nolint register_mlr3() + assign("lg", lgr::get_logger("mlr3spatiotempcv"), envir = parent.env(environment())) setHook(packageEvent("mlr3", "onLoad"), function(...) register_mlr3(), action = "append") }