File tree Expand file tree Collapse file tree 1 file changed +17
-8
lines changed Expand file tree Collapse file tree 1 file changed +17
-8
lines changed Original file line number Diff line number Diff line change 11# Surv-TCAV on PBC-276
22
33End-to-end pipeline for XGBoost AFT on PBC-276 with TreeSHAP and concept-based Surv-TCAV.
4- Figures are written to ` figures/ ` .
4+ Figures are written to ` figures/ ` , metrics to ` results/ ` .
55
66## Environment
77``` bash
@@ -20,16 +20,25 @@ Command-line options:
2020python pbc_surv_tcav.py --help
2121```
2222
23+ ### Faster run (for CI/CODECHECK)
24+ ``` bash
25+ python pbc_surv_tcav.py --repeats 5 --no_shap
26+ ```
27+
2328## Outputs
24- - Console: 25× 80/20 validation, C-index and IBS summary, Surv-TCAV effects
25- - Figures:
26- - ` figures/validation_metrics.png `
27- - ` figures/treeshap_summary.png `
28- - ` figures/tcav_bar.png `
29+ - ** Console** : 25× 80/20 validation, C-index and IBS summary, Surv-TCAV effects
30+ - ** Results (CSV)** :
31+ - results/split_metrics.csv
32+ - results/summary_metrics.csv
33+ - ** Figures** :
34+ - figures/validation_metrics.png
35+ - figures/treeshap_summary.png (optional; disable with --no_shap)
36+ - figures/tcav_bar.png
37+ - figures/calibration_tstar.png
2938
3039## Data
31- PBC dataset mirrored via Rdatasets (Therneau, survival package):
32- https://vincentarelbundock.github.io/Rdatasets/csv/survival/pbc.csv
40+ PBC dataset mirrored via Rdatasets (Therneau, * survival* package):
41+ < https://vincentarelbundock.github.io/Rdatasets/csv/survival/pbc.csv >
3342
3443## License
3544MIT
You can’t perform that action at this time.
0 commit comments