AmayaGS commited on
Commit
4c446c7
·
verified ·
1 Parent(s): 410e87f

Update model card

Browse files
Files changed (1) hide show
  1. README.md +65 -3
README.md CHANGED
@@ -1,3 +1,65 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - medical-imaging
5
+ - computational-pathology
6
+ - survival-analysis
7
+ - multimodal
8
+ - tcga
9
+ datasets:
10
+ - TCGA
11
+ library_name: pytorch
12
+ ---
13
+
14
+ # ProtoPathway
15
+
16
+ Pretrained checkpoints, preprocessed cohort data, and the curated pathway
17
+ graph for **ProtoPathway**, an interpretable-by-design multimodal framework
18
+ for cancer survival prediction.
19
+
20
+ See the [code repository](https://github.com/AmayaGS/ProtoPathway) for usage,
21
+ training, and evaluation instructions.
22
+
23
+ ## Layout
24
+
25
+ ```
26
+ pathways/ shared curated Reactome + Hallmark pathways
27
+ cohorts/{cohort}/
28
+ gene_expression.csv preprocessed expression matrix
29
+ bipartite_graph.pt cohort-specific gene-pathway graph
30
+ labels.csv survival times, events, and bins
31
+ data_splits.pkl 5-fold CV splits (SurvPath-compatible)
32
+ checkpoints/best_fold_{0..4}.pt trained model weights
33
+ ```
34
+
35
+ ## Cohorts
36
+
37
+ Five TCGA cohorts: BRCA (N=714), BLCA (N=359), COADREAD (N=227),
38
+ HNSC (N=392), STAD (N=318). Gene expression is the preprocessed
39
+ SurvPath release. WSI patch features (UNI2-h) are not redistributed
40
+ here and should be obtained from the
41
+ [Mahmood Lab](https://huggingface.co/MahmoodLab/UNI2-h) directly.
42
+
43
+ ## Quick load
44
+
45
+ ```python
46
+ from huggingface_hub import snapshot_download
47
+
48
+ # Everything for one cohort plus the shared pathway file
49
+ snapshot_download(
50
+ repo_id="AmayaGS/ProtoPathway",
51
+ local_dir="./assets",
52
+ allow_patterns=["cohorts/TCGA-BLCA/*", "pathways/*"],
53
+ )
54
+ ```
55
+
56
+ ## Citation
57
+
58
+ ```bibtex
59
+ @article{protopathway2026,
60
+ title = {ProtoPathway: Interpretable Multimodal Cancer Survival Prediction
61
+ via Prototype-Pathway Cross-Modal Attention},
62
+ author = {...},
63
+ year = {2026}
64
+ }
65
+ ```