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 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63
|
#' @export
levels.train <- function(x, ...) {
if(any(names(x) == "levels")) {
out <- x$levels
attributes(out) <- NULL
} else {
if(x$modelType == "Classification") {
if(!isS4(x$finalModel) && !is.null(x$finalModel$obsLevels))
return(x$finalModel$obsLevels)
if(is.null(x$modelInfo)) {
code <- getModelInfo(x$method, regex = FALSE)[[1]]
} else code <- x$modelInfo
if(!is.null(code$levels)){
checkInstall(code$library)
for(i in seq(along.with = code$library))
do.call("requireNamespaceQuietStop", list(package = code$library[i]))
out <- code$levels(x$finalModel, ...)
} else out <- NULL
} else out <- NULL
}
out
}
|