scripts/rforest_R2_RMSE_script.R

In iaolier/mtl-qsar:

### rforest R2 and RMSE ###

output <- data.frame()

all_data <- list.files(path="/shared/mtl-qsar/predictions/stl/rforest")


list_data = list.files("/shared/mtl-qsar/predictions/stl/rforest", 
                       full.names = TRUE)

sz = length(list_data)


for(i in 1:sz){
  data_name = list_data[i]
  preds_dset = read.csv(data_name) # reads in each file of data
  
  print(i)
  
  data_id = all_data[i]
  algorithm = 'RF'
 
  for(foldi in 1:10){
    
    tmp <- filter(preds_dset, foldi == fold)
    
    rmse <- RMSE(tmp$truth,tmp$prediction)
    r2 <- R2(tmp$truth,tmp$prediction)
    fold = foldi
    outp = cbind(data_id, fold, algorithm, rmse, r2)
    output = rbind(output, outp)
   
  }
  
  out.p = as.data.frame(output)
  write.csv(out.p, file="rforest_R2_RMSE.csv")
}