Fixed bug in h2o.ensemble .make_Z function

h2oai · Aug 14, 2015 · 6851fa4 · 6851fa4
1 parent b79f75a
commit 6851fa4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/h2o-r/ensemble/h2oEnsemble-package/R/ensemble.R b/h2o-r/ensemble/h2oEnsemble-package/R/ensemble.R
@@ -176,6 +176,7 @@ h2o.ensemble <- function(x, y, training_frame,
                     seed = seed, fold_column = "fold_id", 
                     simplify = FALSE)
 
+  runtime <- list()
   runtime$cv <- lapply(fitlist, function(ll) ll$fittime)
   names(runtime$cv) <- learner
   basefits <- lapply(fitlist, function(ll) ll$fit)  #Base fits (trained on full data) to be saved