cv_survdnn {survdnn} | R Documentation |
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,
...
)
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()]. |
Value
A tibble containing metric values per fold and (optionally) per time point.
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
)
[Package survdnn version 0.6.0 Index]