Surv-TCAV on PBC-276

End-to-end pipeline for XGBoost AFT on PBC-276 with TreeSHAP and concept-based Surv-TCAV.
Figures are written to figures/, metrics to results/.

Environment

python -m venv .venv
. .venv/bin/activate      # Windows: .venv\Scripts\activate
pip install -r requirements.txt

Run

python pbc_surv_tcav.py

Command-line options:

python pbc_surv_tcav.py --help

Faster run (for CI/CODECHECK)

python pbc_surv_tcav.py --repeats 5 --no_shap

Outputs

• Console: 25× 80/20 validation, C-index and IBS summary, Surv-TCAV effects
• Results (CSV):
  • results/split_metrics.csv
  • results/summary_metrics.csv
• Figures:
  • figures/validation_metrics.png
  • figures/treeshap_summary.png (optional; disable with --no_shap)
  • figures/tcav_bar.png
  • figures/calibration_tstar.png

Data

PBC dataset mirrored via Rdatasets (Therneau, survival package):
https://vincentarelbundock.github.io/Rdatasets/csv/survival/pbc.csv

License

MIT