R/ChainModel.R

In mlr-org/mlr: Machine Learning in R

#' Only exported for internal use.
#' @param next.model ([WrappedModel])\cr
#'   The next model.
#' @param cl ([character])\cr
#'   Subclass to assign to the resulting model.
#'
#' @keywords internal
#' @export
makeChainModel = function(next.model, cl) {
  setClasses(list(next.model = next.model), c(cl, "ChainModel", "WrappedModel"))
}


#' @export
getLearnerModel.BaseWrapperModel = function(model, more.unwrap = FALSE) {
  # FIXME: this structure and special-cases really suck. FailureModel and NoFeaturesModel
  # should probably be redesigned at some point
  if (inherits(model$learner.model, "NoFeaturesModel")) {
    return(model$learner.model)
  }
  if (more.unwrap) {
    getLearnerModel(model$learner.model$next.model, more.unwrap = TRUE)
  } else {
    model$learner.model$next.model
  }
}

#' @export
print.ChainModel = function(x, ...) {
  print(x$next.model)
}