Skip to content

Commit 5e46899

Browse files
authored
Update README.md
1 parent 61f90d4 commit 5e46899

File tree

1 file changed

+17
-8
lines changed

1 file changed

+17
-8
lines changed

README.md

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Surv-TCAV on PBC-276
22

33
End-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:
2020
python 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
3544
MIT

0 commit comments

Comments
 (0)