R/as_learner.R

In mlr3: Machine Learning in R - Next Generation

#' @title Convert to a Learner
#'
#' @description
#' Convert object to a [Learner] or a list of [Learner].
#'
#' @inheritParams as_task
#'
#' @return [Learner].
#' @export
as_learner = function(x, ...) { # nolint
  UseMethod("as_learner")
}

#' @export
#' @rdname as_learner
as_learner.Learner = function(x, clone = FALSE, ...) { # nolint
  if (isTRUE(clone)) x$clone(deep = TRUE) else x
}

#' @export
#' @rdname as_learner
as_learners = function(x, ...) { # nolint
  UseMethod("as_learners")
}

#' @export
#' @rdname as_learner
as_learners.default = function(x, ...) { # nolint
  list(as_learner(x, ...))
}

#' @export
#' @rdname as_learner
as_learners.list = function(x, ...) { # nolint
  lapply(x, as_learner, ...)
}