1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
#' 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)
}
|
