h2o (version 3.0.0.30)

h2o.mse: Retrieves Mean Squared Error Value

Description

Retrieves the mean squared error value from an H2OModelMetrics object.

Usage

h2o.mse(object, valid = FALSE, ...)

Arguments

object

An H2OModelMetrics object of the correct type.

valid

Retreive the validation metric.

...

Extra arguments to be passed if `object` is of type H2OModel (e.g. train=TRUE)

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

library(h2o)
h2o.init()

prosPath <- system.file("extdata", "prostate.csv", package="h2o")
hex <- h2o.uploadFile(prosPath)

hex[,2] <- as.factor(hex[,2])
model <- h2o.gbm(x = 3:9, y = 2, training_frame = hex, distribution = "bernoulli")
perf <- h2o.performance(model, hex)
h2o.mse(perf)