predict.h2o4gpu_model: Make Predictions using Trained H2O4GPU Estimator

In h2o4gpu: Interface to 'H2O4GPU'

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 documentation built on May 18, 2021, 1:07 a.m.