R/gbmwrap1.R

In marcog77/marcoUtils:

#' Fit gbm models using a loop give a set of responses
#'
#' @resp = list of responses
#' @pred= list of predictors
#' @modsave = path of folder where to save models
#' @indat = input dataset
#' 

gbmwrap1<-function(resp,pred,modsave=NULL,indat,family){
    results<-NULL
        for (x in 1:length(resp)){
        mod<-gbm.step(data=indat,gbm.x=match(pred,names(indat)),gbm.y=match(resp[x],
        names (indat)),family=family,tree.complexity = 3,learning.rate=0.001,bag.fraction =0.5,step.size=50,
        plot.main=FALSE)
        filep<-paste(modsave,"/",resp[x],sep="")
        save(mod,file=filep)
        df<-data.frame(summary(mod,plotit=FALSE),variable=resp[x])
        row.names(df)<-1:dim(df)[1]
        results<-rbind(df,results)
    }
    return(results)
}