predict.cuda_ml_fil {cuda.ml}R Documentation

Make predictions on new data points.

Description

Make predictions on new data points using a FIL model.

Usage

## S3 method for class 'cuda_ml_fil'
predict(object, x, output_class_probabilities = FALSE, ...)

Arguments

object

A trained CuML model.

x

A matrix or dataframe containing new data points.

output_class_probabilities

Whether to output class probabilities. NOTE: setting output_class_probabilities to TRUE is only valid when the model being applied is a classification model and supports class probabilities output. CuML classification models supporting class probabilities include knn, fil, and rand_forest. A warning message will be emitted if output_class_probabilities is set to TRUE or FALSE but the model being applied does not support class probabilities output.

...

Additional arguments to predict(). Currently unused.

Value

Predictions on new data points.


[Package cuda.ml version 0.3.2 Index]