Description Usage Arguments Details Author(s) References Examples
View source: R/rsfes.predict.R
Prediction with new data and return a saved forest with mean hazard function
1 | rsfes.predict(rsfesfit, newdata, trlength = 500)
|
rsfesfit |
|
newdata |
|
trlength |
Prediction with new data and return a saved forest with mean hazard function
Hong Wang
Random Survival Forest with Space Extensions for Censored Data, submitted to Artificial Intelligence in Medicine
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 | ##---- Should be DIRECTLY executable !! ----
##-- ==> Define data, use random,
##-- or do help(data=index) for the standard data sets.
## The function is currently defined as
function (rsfesfit, newdata, trlength = 500)
{
trees = rsfesfit$pectrees
colindexes = rsfesfit$colindexes
newindexes = rsfesfit$newindexes
if (trlength > length(rsfesfit$pectrees))
stop("Number of Trees for prediction should not be more than Number of Trees Fitted")
testpre <- NULL
for (i in 1:trlength) {
{
if (ncol(newdata) <= 100) {
testdata = extspace_testdat(newdata, newindexes[[i]])
testdata = testdata[, colindexes[[i]]]
}
else {
testdata = newdata[, colindexes[[i]]]
testdata = extspace_testdat(testdata, newindexes[[i]])
}
testdata = as.data.frame(testdata)
predicts <- predict(trees[[i]]$rpart, testdata)
testpre <- cbind(predicts, testpre)
}
}
ensemble_predictions <- rowMeans(testpre)
return(ensemble_predictions)
}
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.