Retrieves the root mean squared error value from an H2OModelMetrics object. If "train", "valid", and "xval" parameters are FALSE (default), then the training RMSEvalue is returned. If more than one parameter is set to TRUE, then a named vector of RMSEs are returned, where the names are "train", "valid" or "xval".
h2o.rmse(object, train = FALSE, valid = FALSE, xval = FALSE)
| object | An H2OModelMetrics object of the correct type. |
|---|---|
| train | Retrieve the training RMSE |
| valid | Retrieve the validation RMSE |
| xval | Retrieve the cross-validation RMSE |
This function only supports H2OBinomialMetrics, H2OMultinomialMetrics, and H2ORegressionMetrics objects.
h2o.auc for AUC, h2o.mse for RMSE, and
h2o.metric for the various threshold metrics. See
h2o.performance for creating H2OModelMetrics objects.
# NOT RUN { library(h2o) h2o.init() prostate_path <- system.file("extdata", "prostate.csv", package = "h2o") prostate <- h2o.uploadFile(prostate_path) prostate[, 2] <- as.factor(prostate[, 2]) model <- h2o.gbm(x = 3:9, y = 2, training_frame = prostate, distribution = "bernoulli") perf <- h2o.performance(model, prostate) h2o.rmse(perf) # }