hai_kmeans_automl_predict: Automatic K-Means H2O

In healthyR.ai: The Machine Learning and AI Modeling Companion to 'healthyR'

Automatic K-Means H2O

Description

This is a wrapper around the h2o::h2o.predict() function that will return a list object with a lot of useful and easy to use tidy style information.

Usage

hai_kmeans_automl_predict(.input)

Arguments

.input

This is the output of the hai_kmeans_automl() function.

Details

This function will internally take in the output assigned from the hai_kmeans_automl() function only and return a list of useful information. The items that are returned are as follows:

1. prediction - The h2o dataframe of predictions

2. prediction_tbl - The h2o predictions in tibble format

3. valid_tbl - The validation data in tibble format

4. pred_full_tbl - The entire validation set with the predictions attached using base::cbind(). The predictions are in a column called predicted_cluster and are in the formate of a factor using forcats::as_factor()

Value

A list object

Author(s)

Steven P. Sanderson II, MPH

See Also

Other Kmeans: hai_kmeans_automl(), hai_kmeans_mapped_tbl(), hai_kmeans_obj(), hai_kmeans_scree_data_tbl(), hai_kmeans_scree_plt(), hai_kmeans_tidy_tbl(), hai_kmeans_user_item_tbl()

Examples

## Not run: 
h2o.init()

output <- hai_kmeans_automl(
  .data = iris,
  .predictors = c("Sepal.Width", "Sepal.Length", "Petal.Width", "Petal.Length"),
  .standardize = FALSE
)

pred <- hai_kmeans_automl_predict(output)

h2o.shutdown()

## End(Not run)

healthyR.ai documentation built on April 3, 2023, 5:24 p.m.