select_best_hyperparameters: Select the best hyperparameter set according to a given...

Description Usage Arguments Value

View source: R/hyperparameter-tuning.R

Description

Metaflow saves artefacts as pickle files, which prevents us from using nested tibbles. Unfortunately, tune relies on nested tibbles to store its results. We instead use collect_metrics to store simple tibbles that are compatible with pickle. Unfortunately, this means we cannot rely on tune's functions for extracting optimal hyperparameter sets and must write our own.

Usage

1
select_best_hyperparameters(hyperparameter_results, metric = "rmse")

Arguments

hyperparameter_results

A tibble of hyperparameter results collected with tune::collect_metrics

metric

A metric to minimise. Defaults to "rmse".

Value

named list


mdneuzerling/YelpMetaflow documentation built on Jan. 25, 2021, 12:36 a.m.