rsfes.surv_predict: Prediction with new data and return a saved forest with mean...

In whcsu/sfse: Random Survival Forest with Space Extensions (short line)

Description

Prediction with new data and return a saved forest with mean surv prob at each time points

Usage

rsfes.surv_predict(rsfesfit, newdata, uniquetimes, trlength = 500)

Arguments

rsfesfit
newdata
uniquetimes
trlength

Details

Prediction with new data and return a saved forest with mean surv prob at each time points

Author(s)

Hong Wang

References

Random Survival Forest with Space Extensions for Censored Data, submitted to Artificial Intelligence in Medicine

Examples

##---- 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, uniquetimes, 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 <- matrix(0, nrow = dim(newdata)[1], ncol = length(uniquetimes))
    colnames(testpre) = paste0(uniquetimes)
    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 = predictSurvProb(trees[[i]], testdata, 
                uniquetimes)
            predicts[is.na(predicts)] = 0
            colnames(predicts) = paste0(uniquetimes)
            testpre <- testpre + predicts
        }
    }
    return(testpre/trlength)
  }

whcsu/sfse documentation built on Dec. 4, 2019, 2:11 p.m.