SHAPBoost (version 1.0.3)

SHAPBoostSurvival-class: SHAPBoostSurvival is a reference class for survival analysis feature selection through gradient boosting.

Description

This class extends the SHAPBoostEstimator class and implements methods for initializing, updating weights, scoring, and fitting estimators.

Fields

evaluator

The model that is used to evaluate each additional feature. Choice between "coxph" and "xgb".

metric

The metric used for evaluation, such as "mae", "mse", or "r2".

xgb_params

A list of parameters for the XGBoost model.

number_of_folds

The number of folds for cross-validation.

epsilon

A small value to prevent division by zero.

max_number_of_features

The maximum number of features to consider.

siso_ranking_size

The size of the SISO ranking.

siso_order

The order of the SISO ranking.

reset

A boolean indicating whether to reset the model.

xgb_importance

The importance type for XGBoost.

num_resets

The number of resets for the model.

fold_random_state

The random state for folds.

verbose

The verbosity level for logging.

stratification

A boolean indicating whether to use stratification. Only applicable for c-index metric.

use_shap

A boolean indicating whether to use SHAP values.

collinearity_check

A boolean indicating whether to check for collinearity.

correlation_threshold

The threshold for correlation to consider features as collinear.

Examples

if (requireNamespace("survival", quietly = TRUE)) {
  shapboost <- SHAPBoostSurvival$new(
    max_number_of_features = 1,
    evaluator = "coxph",
    metric = "c-index",
    verbose = 0,
    xgb_params = list(
      objective = "survival:cox",
      eval_metric = "cox-nloglik"
    )
  )
  
  X <- as.data.frame(survival::gbsg[, -c(1, 10, 11)])
  y <- as.data.frame(survival::gbsg[, c(10, 11)])
  subset <- shapboost$fit(X, y)
}