daylight-00 commited on
Commit
a9c8ec9
·
verified ·
1 Parent(s): 70f8f67

Cite the posted bioRxiv preprint (doi 10.64898/2026.09.15.751749), and sync the card with the canonical copy

Browse files
Files changed (1) hide show
  1. README.md +16 -15
README.md CHANGED
@@ -47,9 +47,10 @@ fifteen — the one with the lowest validation loss, the value in the table abov
47
  quantity took part in the choice. For a benchmark comparison, use the paper's means, not one
48
  checkpoint.
49
 
50
- Quote the validation losses from `analysis/val_metrics.csv` in the repository. The `val_loss` and
51
- `val_auc` columns of `analysis/scoring/*/per_run_results.csv` are not usable for this: the first
52
- predates a parser fix, and the second comes from a `roc_auc_score` call with its arguments swapped.
 
53
 
54
  Held-out test sets, never used for selection: 48,352 rows (qualitative), 33,490 (ms), 14,150 (both
55
  IC50 arms).
@@ -66,12 +67,9 @@ From the IEDB Export v3, split into arms by measurement type. The rebuild is
66
  | ic50_500 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_500` | 36.7 % |
67
  | ic50_1000 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_1000` | 46.0 % |
68
 
69
- The qualitative arm's training config names `dataset/full_bal/`, a directory since renamed to
70
- `full`. Same CSVs.
71
-
72
  Training: AdamW, lr 1e-5, betas (0.9, 0.999), eps 1e-8, weight decay 0.01, five-fold CV, best epoch
73
- by validation loss (20, 27, 28, 28 for the four files above — the epoch each file stores, and the
74
- corrected argmin of its run). Both sides were fed from precomputed
75
  ESMC 300M embeddings, `emb_hla_esmc_small_0430.h5` for the HLA chains and the matching epitope
76
  store.
77
 
@@ -118,15 +116,15 @@ Two limits worth knowing before feeding it anything:
118
 
119
  ```bash
120
  git clone https://github.com/daylight-00/PREpiBind && cd PREpiBind
121
- pip install --extra-index-url https://download.pytorch.org/whl/cu126 -e .
122
 
123
  hf download daylight-00/prepibind prepibind_qualitative_s100_f0.pt --local-dir models
124
  hf download daylight-00/esmc-300m-2024-12 esmc_300m_2024_12_v0_fp16.pth --local-dir models
125
  hf download daylight-00/prepibind-embeddings emb_hla_esmc_small_0430.h5 --repo-type dataset --local-dir emb
126
  ```
127
 
128
- The ESMC encoder is vendored into the package (`prepibind/esmc/`), so `torch` and
129
- `huggingface_hub` are the only runtime dependencies for inference — no `pip install esm`.
130
 
131
  ```python
132
  from prepibind.inference import load_config, main
@@ -155,8 +153,9 @@ The CLI does the same thing for the configs' own settings:
155
  python -m prepibind.inference configs/predict/config_ms.py --chkp_path models/prepibind_ms_s128_f3.pt --plot
156
  ```
157
 
158
- No GPU is required. Without CUDA, or on a pre-Ampere card, the runtime turns off flash-attn, falls
159
- back to float32 and says so.
 
160
 
161
  ## Limitations and known caveats
162
 
@@ -191,8 +190,10 @@ licence. Training data derives from IEDB (CC BY 4.0) and allele sequences from I
191
  title = {PREpiBind: Protein Representation-integrated Epitope-MHC Class II Binding Prediction},
192
  author = {Jang, David Hyunyoo and Kim, Dongwoo and Park, Byungho and Hwang, Untaek and Choi, Yoonjoo and Lee, Juyong},
193
  journal = {bioRxiv},
194
- year = {2026}
 
 
195
  }
196
  ```
197
 
198
- The preprint is not posted yet; the DOI goes here when it is.
 
47
  quantity took part in the choice. For a benchmark comparison, use the paper's means, not one
48
  checkpoint.
49
 
50
+ Quote the validation losses from `analysis/val_metrics.csv` in the repository; the `val_loss`
51
+ column of `analysis/scoring/*/per_run_results.csv` is joined in from that file and carries the same
52
+ numbers. The `val_auc` column in either is not usable: it comes from a `roc_auc_score` call with its
53
+ arguments swapped.
54
 
55
  Held-out test sets, never used for selection: 48,352 rows (qualitative), 33,490 (ms), 14,150 (both
56
  IC50 arms).
 
67
  | ic50_500 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_500` | 36.7 % |
68
  | ic50_1000 | `data/dataset/ic50/` | 33,004 | 14,150 | `Target_1000` | 46.0 % |
69
 
 
 
 
70
  Training: AdamW, lr 1e-5, betas (0.9, 0.999), eps 1e-8, weight decay 0.01, five-fold CV, best epoch
71
+ by validation loss (20, 27, 28, 28 for the four files above — the epoch each source checkpoint
72
+ recorded, and the corrected argmin of its run). Both sides were fed from precomputed
73
  ESMC 300M embeddings, `emb_hla_esmc_small_0430.h5` for the HLA chains and the matching epitope
74
  store.
75
 
 
116
 
117
  ```bash
118
  git clone https://github.com/daylight-00/PREpiBind && cd PREpiBind
119
+ pip install --extra-index-url https://download.pytorch.org/whl/cu126 -e ".[demo]"
120
 
121
  hf download daylight-00/prepibind prepibind_qualitative_s100_f0.pt --local-dir models
122
  hf download daylight-00/esmc-300m-2024-12 esmc_300m_2024_12_v0_fp16.pth --local-dir models
123
  hf download daylight-00/prepibind-embeddings emb_hla_esmc_small_0430.h5 --repo-type dataset --local-dir emb
124
  ```
125
 
126
+ The ESMC encoder is vendored into the package (`prepibind/esmc/`) and runs on `torch` alone, so
127
+ inference needs no `pip install esm`.
128
 
129
  ```python
130
  from prepibind.inference import load_config, main
 
153
  python -m prepibind.inference configs/predict/config_ms.py --chkp_path models/prepibind_ms_s128_f3.pt --plot
154
  ```
155
 
156
+ No GPU is required, and the runtime says whatever it had to drop: without CUDA, float32 throughout
157
+ and no flash-attn; on a pre-Ampere card, flash-attn off and the bfloat16 ESMC demoted to float16,
158
+ head unchanged.
159
 
160
  ## Limitations and known caveats
161
 
 
190
  title = {PREpiBind: Protein Representation-integrated Epitope-MHC Class II Binding Prediction},
191
  author = {Jang, David Hyunyoo and Kim, Dongwoo and Park, Byungho and Hwang, Untaek and Choi, Yoonjoo and Lee, Juyong},
192
  journal = {bioRxiv},
193
+ year = {2026},
194
+ doi = {10.64898/2026.09.15.751749},
195
+ url = {https://www.biorxiv.org/content/10.64898/2026.09.15.751749v1}
196
  }
197
  ```
198
 
199
+ Posted on bioRxiv 2026-09-21 as [doi:10.64898/2026.09.15.751749](https://doi.org/10.64898/2026.09.15.751749), CC BY 4.0.