predict.h2o4gpu_model {h2o4gpu}

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)