R/predictions.R

In koenvanbenthem/JCRForest: Fitting a combined regression/classification random forest

predict.jcr_forest <- function(forest,newdata){
  # tests:
  # 1. number of columns
  # possibly: is x numeric?
  nsample <- nrow(newdata)
  
  newdata <- data.matrix(newdata)
  
  pred.out <- .C("forestpred",
                 x=newdata,
                 yc=numeric(nsample),
                 yf=matrix(integer(nsample * forest$nclass),nrow=forest$nclass),
                 nclass=forest$nclass,
                 nsample=as.integer(nsample),
                 ldaughter=forest$ldaughter,
                 rdaughter=forest$rdaughter,
                 yf_pred=forest$yf_pred,
                 yc_pred=forest$yc_mu_pred,
                 ntree = as.integer(forest$ntree),
                 nrnodes = as.integer(forest$nrnodes),
                 node_var = forest$node_var,
                 node_xvar = forest$node_xvar,
                 PACKAGE="JCRForest"
  )
  return(pred.out)
}