h2o.cross_validation_predictions: Retrieve the cross-validation predictions

In h2o: R Interface for the 'H2O' Scalable Machine Learning Platform

Retrieve the cross-validation predictions

Description

Retrieve the cross-validation predictions

Usage

h2o.cross_validation_predictions(object)

Arguments

object

An H2OModel object.

Value

Returns a list of H2OFrame objects

Examples

## Not run: 
library(h2o)
h2o.init()

f <- "https://s3.amazonaws.com/h2o-public-test-data/smalldata/junit/cars_20mpg.csv"
cars <- h2o.importFile(f)
cars["economy_20mpg"] <- as.factor(cars["economy_20mpg"])
predictors <- c("displacement", "power", "weight", "acceleration", "year")
response <- "economy_20mpg"
cars_split <- h2o.splitFrame(data = cars, ratios = 0.8, seed = 1234)
train <- cars_split[[1]]
valid <- cars_split[[2]]
cars_gbm <- h2o.gbm(x = predictors, y = response, training_frame = train, 
                    nfolds = 5,  keep_cross_validation_predictions = TRUE, seed = 1234)
h2o.cross_validation_predictions(cars_gbm)

## End(Not run)

h2o documentation built on May 29, 2024, 4:26 a.m.