get_pycox_callbacks: Get Torchtuples Callbacks

In RaphaelS1/survivalmodels: Models for Survival Analysis

Get Torchtuples Callbacks

Description

Helper function to return torchtuples callbacks from torchtuples.callbacks.

Usage

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

if (requireNamespaces("reticulate")) {
  get_pycox_callbacks(early_stopping = TRUE)

  get_pycox_callbacks(best_weights = TRUE)
}

RaphaelS1/survivalmodels documentation built on July 5, 2024, 10:17 a.m.