Retrieves the mean squared error value from an H2OModelMetrics object. If "train", "valid", and "xval" parameters are FALSE (default), then the training MSEvalue is returned. If more than one parameter is set to TRUE, then a named vector of MSEs are returned, where the names are "train", "valid" or "xval".
h2o.mse(object, train = FALSE, valid = FALSE, xval = FALSE)
object | An H2OModelMetrics object of the correct type. |
---|---|
train | Retrieve the training MSE |
valid | Retrieve the validation MSE |
xval | Retrieve the cross-validation MSE |
This function only supports H2OBinomialMetrics, H2OMultinomialMetrics, and H2ORegressionMetrics objects.
h2o.auc
for AUC, h2o.mse
for MSE, 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.mse(perf) # }