Get Torchtuples Callbacks — get_pycox_callbacks • survivalmodels

Helper function to return torchtuples callbacks from torchtuples.callbacks.

get_pycox_callbacks(
  early_stopping = FALSE,
  best_weights = FALSE,
  min_delta = 0,
  patience = 10L
)

Arguments

early_stopping: (logical(1))
If TRUE then constructs torchtuples.callbacks,EarlyStopping.
best_weights: (logical(1))
If TRUE then returns torchtuples.callbacks.BestWeights. Ignored if early_stopping is TRUE.
min_delta: (numeric(1))
Passed to torchtuples.callbacks.EarlyStopping.
patience: (integer(1))
Passed to torchtuples.callbacks.EarlyStopping.

Examples

# \donttest{
if (requireNamespaces("reticulate")) {
  get_pycox_callbacks(early_stopping = TRUE)

  get_pycox_callbacks(best_weights = TRUE)
}
#> Error in py_module_import(module, convert = convert): ModuleNotFoundError: No module named 'torchtuples'
#> Run `reticulate::py_last_error()` for details.
# }