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
|
varUsed <- function(x, by.tree=FALSE, count=TRUE) {
if (!inherits(x, "randomForest"))
stop(deparse(substitute(x)), "is not a randomForest object")
if (is.null(x$forest))
stop(deparse(substitute(x)), "does not contain forest")
p <- length(x$forest$ncat) # Total number of variables.
if (count) {
if (by.tree) {
v <- apply(x$forest$bestvar, 2, function(x) {
xx <- numeric(p)
y <- table(x[x>0])
xx[as.numeric(names(y))] <- y
xx
})
} else {
v <- numeric(p)
vv <- table(x$forest$bestvar[x$forest$bestvar > 0])
v[as.numeric(names(vv))] <- vv
}
} else {
v <- apply(x$forest$bestvar, 2, function(x) sort(unique(x[x>0])))
if(!by.tree) v <- sort(unique(unlist(v)))
}
v
}
|
