survdnn (version 0.6.0)

cv_survdnn: K-Fold Cross-Validation for survdnn Models

Description

Performs cross-validation for a `survdnn` model using the specified evaluation metrics.

Usage

cv_survdnn(
  formula,
  data,
  times,
  metrics = c("cindex", "ibs"),
  folds = 5,
  .seed = NULL,
  ...
)

Value

A tibble containing metric values per fold and (optionally) per time point.

Arguments

formula

A survival formula, e.g., `Surv(time, status) ~ x1 + x2`.

data

A data frame.

times

A numeric vector of evaluation time points.

metrics

A character vector: any of `"cindex"`, `"brier"`, `"ibs"`.

folds

Integer. Number of folds to use.

.seed

Optional. Set random seed for reproducibility.

...

Additional arguments passed to [survdnn()].

Examples

library(survival)
data(veteran)
cv_survdnn(
  Surv(time, status) ~ age + karno + celltype,
  data = veteran,
  times = c(30, 90, 180),
  metrics = "ibs",
  folds = 3,
  .seed = 42,
  hidden = c(16, 8),
  epochs = 5
)