healthyR.ai (version 0.1.0)

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

Value

A list object

Arguments

.input

This is the output of the hai_kmeans_automl() function.

Author

Steven P. Sanderson II, MPH

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

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

if (FALSE) {
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()
}