h2o4gpu (version 0.2.0)

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 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)

# }
# NOT RUN {
# }