R/predict.rf.R

In cugliari/enercast: Electrical energy forecast tools

#' predict.rf
#'
#' Prediction in Random Forest
#' 
#' @param listRF List of Random Forest Models
#' @param newdata New data to predict
#' @param for.each Character. Variable name of data frequency (i.e. 'hour')
#' 
#' @return list
#' @author Jairo Cugliari, Andres Castrillejo, Fernando Massa, Ignacio Ramirez
predict.rf <- function(listRF, newdata=NULL, for.each='hour'){
  if(is.null(newdata)){
    PRED.RF <- lapply(listRF,predict)}
  if(!is.null(newdata)){
    k <- which(names(newdata)%in%for.each)
    PRED.RF <- NULL
    for(i in 1:nrow(newdata)){
      wrf <- which(names(listRF)%in%newdata[i,k])
      pred.rf <- predict(listRF[[wrf]],newdata[i,])
      PRED.RF <- c(PRED.RF,pred.rf)
    }
  }
  return(PRED.RF)
}