Skip to content

Commit

Permalink
Remove sanity/bounds check on predicted values
Browse files Browse the repository at this point in the history
  • Loading branch information
billderose committed Jan 14, 2015
1 parent d957efb commit b95f042
Showing 1 changed file with 0 additions and 5 deletions.
5 changes: 0 additions & 5 deletions R/ensemble/h2oEnsemble-package/R/ensemble.R
Original file line number Diff line number Diff line change
Expand Up @@ -224,8 +224,6 @@ predict.h2o.ensemble <- function(object, newdata) {
}
}
names(basepreddf) <- names(object$basefits)
basepreddf[basepreddf < object$ylim[1]] <- object$ylim[1] #Enforce bounds
basepreddf[basepreddf > object$ylim[2]] <- object$ylim[2]
basepred <- as.h2o(localH2O, basepreddf, key="basepred")

if (grepl("H2O", class(object$metafit))) {
Expand All @@ -235,9 +233,6 @@ predict.h2o.ensemble <- function(object, newdata) {
# SuperLearner wrapper function metalearner
pred <- predict(object=object$metafit$fit, newdata=basepred)
}
# TO DO: Maybe restrict bounds here, but change code to work for pred as H2OParsedData obj
#pred[pred < object$ylim[1]] <- object$ylim[1] #Enforce bounds
#pred[pred > object$ylim[2]] <- object$ylim[2]
out <- list(pred = pred, basepred = basepred)
return(out)
}
Expand Down

0 comments on commit b95f042

Please sign in to comment.