Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

predict.h2o4gpu_model

Make Predictions using Trained H2O4GPU Estimator


Description

This function makes predictions from new data using a trained H2O4GPU model and returns class predictions for classification and predicted values for regression.

Usage

## S3 method for class 'h2o4gpu_model'
predict(object, x, type = "raw", ...)

Arguments

object

The h2o4gpu model object

x

The new data where each column represents a different predictor variable to be used in generating predictions.

type

One of "raw" or "prob", indicating the type of output: predicted values or probabilities

...

Additional arguments (unused for now).

Examples

## Not run: 

library(h2o4gpu)

# Setup dataset
x <- iris[1:4]
y <- as.integer(iris$Species) - 1

# Initialize and train the classifier
model <- h2o4gpu.random_forest_classifier() %>% fit(x, y)

# Make predictions
predictions <- model %>% predict(x)


## End(Not run)

h2o4gpu

Interface to 'H2O4GPU'

v0.2.0
Apache License 2.0
Authors
Yuan Tang [aut, cre] (<https://orcid.org/0000-0001-5243-233X>), Navdeep Gill [aut], Erin LeDell [aut], H2O.ai [cph, fnd]
Initial release

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.