JiecongLin commited on
Commit
981fbfc
·
verified ·
1 Parent(s): c2dba76

Add model card + K562 fold-8 enhancer encoder

Browse files
Files changed (2) hide show
  1. README.md +62 -0
  2. enhancer_encoders/K562/fold_8.pt +3 -0
README.md ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ tags:
4
+ - biology
5
+ - genomics
6
+ - gene-expression
7
+ - enhancer
8
+ - regulatory-genomics
9
+ - dna
10
+ library_name: pytorch
11
+ ---
12
+
13
+ # EPInformer — reproducible pipeline (pretrained checkpoints)
14
+
15
+ Pretrained checkpoints for the **[EPInformer reproducible pipeline](https://github.com/pinellolab/EPInformer/tree/reproducible_pipeline)** — a from-raw-ENCODE reproduction of [EPInformer](https://github.com/pinellolab/EPInformer) across 6 cell lines (K562, GM12878, H1, HepG2, HUVEC, NHEK).
16
+
17
+ Two models (defined in `EPInformer/models.py` on the `reproducible_pipeline` branch):
18
+
19
+ - **`enhancer_predictor_256bp`** — 256 bp enhancer-activity encoder; predicts `log2(0.1 + sqrt(H3K27ac·DNase))` activity from sequence.
20
+ - **`EPInformer_v2`** — gene-expression model (RNA / CAGE) that reuses the frozen encoder as its sequence backbone.
21
+
22
+ ## Results (12-fold leave-chromosome-out, pooled Pearson R)
23
+
24
+ **Enhancer encoder** (log2 activity):
25
+
26
+ | | H1 | HepG2 | K562 | HUVEC | NHEK | GM12878 |
27
+ |---|---|---|---|---|---|---|
28
+ | **R** | 0.820 | 0.743 | 0.740 | 0.742 | 0.677 | 0.617 |
29
+
30
+ **Gene expression** (shipped `f3` = frozen encoder + 3 enhancer features + promoter signal):
31
+
32
+ | | K562 | GM12878 | HepG2 | HUVEC | NHEK | H1 |
33
+ |---|---|---|---|---|---|---|
34
+ | **RNA** | 0.856 | 0.860 | 0.845 | 0.839 | 0.828 | 0.781 |
35
+ | **CAGE** | 0.867 | 0.890 | — | — | — | — |
36
+
37
+ CAGE labels exist only for K562/GM12878 (the other four are RNA-only).
38
+
39
+ ## Files
40
+
41
+ ```
42
+ enhancer_encoders/{CELL}/fold_{i}.pt # 256 bp enhancer-activity encoders, 12 leave-chromosome-out folds per cell
43
+ ```
44
+
45
+ The fold-8 K562 encoder (`enhancer_encoders/K562/fold_8.pt`) is the checkpoint used in the KLF1 enhancer-activity demo notebook.
46
+
47
+ ## Usage
48
+
49
+ ```python
50
+ import torch
51
+ from huggingface_hub import hf_hub_download
52
+ # the model class lives on the reproducible_pipeline branch:
53
+ # git clone -b reproducible_pipeline https://github.com/pinellolab/EPInformer
54
+ from EPInformer.models import enhancer_predictor_256bp
55
+
56
+ ckpt = hf_hub_download("JiecongLin/EPInformer-reproducible", "enhancer_encoders/K562/fold_8.pt")
57
+ net = enhancer_predictor_256bp()
58
+ net.load_state_dict(torch.load(ckpt, map_location="cpu", weights_only=False)["model_state_dict"])
59
+ net.eval()
60
+ ```
61
+
62
+ See the **[project wiki](https://github.com/pinellolab/EPInformer/wiki)** for the full training and evaluation guide, and the **[`reproducible_pipeline`](https://github.com/pinellolab/EPInformer/tree/reproducible_pipeline)** branch for code.
enhancer_encoders/K562/fold_8.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dbbc604689afa14b064fd3ad4828acd33aa84edd982a451c07eab3b2e46ed8f2
3
+ size 3244547