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".
Arguments
- object
An H2OModelMetrics object of the correct type.
- train
Retrieve the training MSE
- valid
Retrieve the validation MSE
- xval
Retrieve the cross-validation MSE
Details
This function only supports H2OBinomialMetrics, H2OMultinomialMetrics, and H2ORegressionMetrics objects.
See also
h2o.auc for AUC, h2o.mse for MSE, and
h2o.metric for the various threshold metrics. See
h2o.performance for creating H2OModelMetrics objects.
Examples
if (FALSE) { # \dontrun{
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)
} # }