predict.h2o4gpu_model: Make Predictions using Trained H2O4GPU Estimator

Description Usage Arguments Examples

View source: R/model.R

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

1
2
## 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

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
## 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 documentation built on May 18, 2021, 1:07 a.m.