survivalmodels (version 0.1.191)

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

Value

No return value.

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.