vr-scientist commited on
Commit
5312ba5
·
verified ·
1 Parent(s): f5f5ce7

Add BPNet model ENCSR584FHS (ENCSR892DRK)

Browse files
.gitattributes CHANGED
@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ fold_0/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
37
+ fold_1/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
38
+ fold_2/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
39
+ fold_3/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
40
+ fold_4/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,137 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-4.0
3
+ library_name: bpnet
4
+ tags:
5
+ - bpnet
6
+ - dna
7
+ - genomics
8
+ - transcription-factor-binding
9
+ - chip-seq
10
+ - encode
11
+ - encode-bpnet-atlas
12
+ - hg38
13
+ - qc-passed
14
+ - REST
15
+ ---
16
+
17
+ # ENCODE BPNet Atlas
18
+
19
+ As part of the ENCODE 4 Project, we trained BPNet models on 2,339 ENCODE
20
+ transcription factor ChIP-seq experiments spanning 788 targets across
21
+ 175 biosamples. Here, we provide all models for open-source use.
22
+
23
+ For more information about the models, see:
24
+
25
+ - Main ENCODE 4 Paper
26
+ - A unified lexicon of predictive DNA sequence motifs from ENCODE transcription
27
+ factor binding and chromatin accessibility assays (Deshpande et al., Zenodo 2025)
28
+ - Base-resolution models of transcription-factor binding reveal soft motif syntax
29
+ (Avsec et al., Nat Genet 2021)
30
+
31
+ ## BPNet model: REST ChIP-seq in A549 (ENCSR892DRK)
32
+
33
+ - Model: BPNet
34
+ - Assay: TF ChIP-seq
35
+ - Target: REST
36
+ - Experiment: [ENCSR892DRK](https://www.encodeproject.org/experiments/ENCSR892DRK/)
37
+ - Model annotation: [ENCSR584FHS](https://www.encodeproject.org/annotations/ENCSR584FHS/)
38
+ - Biosample: A549 (Full name: Homo sapiens A549)
39
+ - Cell slim(s): cancer cell
40
+ - Organ slim(s): lung
41
+ - Developmental slim(s): endoderm
42
+ - System slim(s): respiratory system
43
+ - Assembly: hg38
44
+
45
+ ## QC
46
+
47
+ - Status: passed
48
+ - Notes: Found direct motif (counts, profile);
49
+
50
+ ## Directory structure
51
+
52
+ 5-fold cross-validation. Each `fold_*/` contains the trained BPNet model in two formats:
53
+
54
+ - `fold_0/model.h5` — BPNet model in .h5 (Keras) format
55
+ - `fold_0/saved_model/` — BPNet model in TensorFlow SavedModel format (a directory; load directly)
56
+ - `config.json` — training / architecture parameters
57
+
58
+ ## Instructions
59
+
60
+ BPNet takes a one-hot DNA sequence plus control (bias) inputs and predicts
61
+ stranded profile logits and total logcounts. The control inputs come from the
62
+ matched WCE/Input DNA control and **can be passed as zeros**.
63
+
64
+ ### 1. Loading the SavedModel and making predictions
65
+
66
+ ```python
67
+ import numpy as np
68
+ import tensorflow as tf
69
+ from scipy.special import logsumexp
70
+
71
+ model = tf.saved_model.load("fold_0/saved_model")
72
+ # sequence: (N, 2114, 4) one-hot [A,C,G,T]
73
+ # profile_bias_input: (N, 1000, 2) per-base profile bias from WCE/Input control, or zeros
74
+ # counts_bias_input: (N, 2) log2 total counts from WCE/Input control, or zeros
75
+ predictions = model.signatures["serving_default"](**{
76
+ "sequence": sequence.astype("float32"),
77
+ "profile_bias_input_0": profile_bias_input.astype("float32"),
78
+ "counts_bias_input_0": counts_bias_input.astype("float32")})
79
+ # predictions["profile_predictions"]: (N, 1000, 2) logits (strands NOT independent)
80
+ # predictions["logcounts_predictions"]: (N, 1) total logcount
81
+
82
+ output_len = 1000
83
+ def vectorized_prediction_to_profile(predictions):
84
+ logits_arr = predictions["profile_predictions"]
85
+ counts_arr = predictions["logcounts_predictions"]
86
+ pred_profile_logits = np.reshape(logits_arr, [-1, 1, output_len * 2])
87
+ probVals_array = np.exp(pred_profile_logits - logsumexp(
88
+ pred_profile_logits, axis=2).reshape([len(logits_arr), 1, 1]))
89
+ profile_predictions = np.multiply(
90
+ np.exp(counts_arr).reshape([len(counts_arr), 1, 1]), probVals_array)
91
+ plus = np.reshape(profile_predictions, [len(counts_arr), output_len, 2])[:, :, 0]
92
+ minus = np.reshape(profile_predictions, [len(counts_arr), output_len, 2])[:, :, 1]
93
+ return plus, minus, counts_arr
94
+
95
+ plus, minus, logcounts = vectorized_prediction_to_profile(predictions)
96
+ ```
97
+
98
+ ### 2. Loading the .h5 (Keras) and making predictions
99
+
100
+ ```python
101
+ import numpy as np
102
+ import tensorflow as tf
103
+ import tensorflow.keras.backend as kb
104
+ from tensorflow.keras.models import load_model
105
+ from tensorflow.keras.utils import CustomObjectScope
106
+ from bpnet.model.custommodel import CustomModel
107
+
108
+ def get_model(model_path):
109
+ with CustomObjectScope({"kb": kb, "tf": tf, "CustomModel": CustomModel}):
110
+ return load_model(model_path)
111
+
112
+ model = get_model("fold_0/model.h5")
113
+ N = sequence.shape[0]
114
+ predictions = model.predict([
115
+ sequence, # (N, 2114, 4)
116
+ np.zeros((N, 1000, 2)), # profile_bias_input (or real WCE/Input control values)
117
+ np.zeros((N, 2))]) # counts_bias_input (or real control log2 counts)
118
+ # predictions[0]: (N, 1000, 2) logits; predictions[1]: (N, 1) logcounts
119
+ # convert with the same vectorized_prediction_to_profile() (predictions[0], predictions[1])
120
+ ```
121
+
122
+ ## Docker image to load and use the models
123
+
124
+ `kundajelab/bpnet-atlas` (placeholder — image forthcoming).
125
+
126
+ ## Code
127
+
128
+ - Code: https://github.com/kundajelab/bpnet/
129
+ - Toolbox & downstream analysis: https://github.com/kundajelab/bpnet/wiki
130
+
131
+ ## License & citation
132
+
133
+ External data users may freely download, analyze and publish results based on any
134
+ ENCODE data without restrictions.
135
+
136
+ Released under the ENCODE data-use policy. Please cite the ENCODE Project
137
+ Consortium and the model software: BPNet (Avsec et al., Nat Genet 2021).
config.json ADDED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "input_len": 2114,
3
+ "output_profile_len": 1000,
4
+ "motif_module_params": {
5
+ "filters": [64],
6
+ "kernel_sizes": [21],
7
+ "padding": "valid"
8
+ },
9
+ "syntax_module_params": {
10
+ "num_dilation_layers": 8,
11
+ "filters": 64,
12
+ "kernel_size": 3,
13
+ "padding": "valid",
14
+ "pre_activation_residual_unit": true
15
+ },
16
+ "profile_head_params": {
17
+ "filters": 1,
18
+ "kernel_size": 75,
19
+ "padding": "valid"
20
+ },
21
+ "counts_head_params": {
22
+ "filters": 1,
23
+ "kernel_size": 75,
24
+ "padding": "valid",
25
+ "units": [1],
26
+ "activations":["linear"],
27
+ "dropouts":[0]
28
+
29
+ },
30
+ "profile_bias_module_params": {
31
+ "kernel_sizes": [1]
32
+ },
33
+ "counts_bias_module_params": {
34
+ },
35
+ "use_attribution_prior": false,
36
+ "attribution_prior_params": {
37
+ "frequency_limit": 150,
38
+ "limit_softness": 0.2,
39
+ "grad_smooth_sigma": 3,
40
+ "profile_grad_loss_weight": 200,
41
+ "counts_grad_loss_weight": 100
42
+ },
43
+ "loss_weights": [1, 251.0742334322453],
44
+ "counts_loss": "MSE"
45
+ }
fold_0/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac49a259acba8cb5c6ee7020918ff75c5e6e659164defebc87583a3ae773747d
3
+ size 561784
fold_0/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d34fbc6911bc8b7a299ad8c0ccaa449e6f7b1c6d79c0c9673c1c5d1cd6246e98
3
+ size 750201
fold_0/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd1528bf90442b3d19f3a15a88f90c1f8dd587ff445afdb24d1dec8857f332a8
3
+ size 1393322
fold_0/saved_model/variables/variables.index ADDED
Binary file (5.77 kB). View file
 
fold_1/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d880d3a7fc224a2f9f6ceb2f878366f6a99c01e230fbb6b3ddf6a209cbce25dc
3
+ size 561784
fold_1/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b46a99ab61203c341d17905a8ab897d82fb6512c8bf5c8c4fd4b97453fe3d91
3
+ size 750201
fold_1/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4a8c1b7918078fd1a44393a71eab3fbb84308916314caeedef3cd3ba55465300
3
+ size 1393322
fold_1/saved_model/variables/variables.index ADDED
Binary file (5.77 kB). View file
 
fold_2/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7e3174cd126626a8e00d5af31cbead4bc05f339047ee0c38d5bdf3019fce3ba9
3
+ size 561784
fold_2/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:19c680db757b6f9ede3a115411987452c5641f37ee3119c0ac350ba3dfad6914
3
+ size 750201
fold_2/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:087b35303ac51dd0dd3f8219c749bf93fc6db2717f58a668b9b6c24ebe292e09
3
+ size 1393322
fold_2/saved_model/variables/variables.index ADDED
Binary file (5.77 kB). View file
 
fold_3/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5fada82d9bf2b042987d2a22bef1413e1788c26889873cc04ddd5dce6a57ebc
3
+ size 561784
fold_3/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:454eaeaac5d8daf8cd2e72a4d94f4c8a1b536110249557006a4f9be7ee7b8d07
3
+ size 750201
fold_3/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c9d29ae1930e61dc4f69231f3c11141c51d5c77d2e5827ad2f6047ec4d17d2b
3
+ size 1393322
fold_3/saved_model/variables/variables.index ADDED
Binary file (5.77 kB). View file
 
fold_4/model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dfc02207092acef98b8e4d43747601a04d00f093b08955555b87cdc725ec8fe6
3
+ size 561784
fold_4/saved_model/saved_model.pb ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:194dd5a2d26e8e1d9e26433a80dc3202d76ad96c1d839bc7ffe77ed247b5f488
3
+ size 750201
fold_4/saved_model/variables/variables.data-00000-of-00001 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c8a80f593ea9eb648147940a036bc56293a50d598d733efac9961ee6c907fb3
3
+ size 1393322
fold_4/saved_model/variables/variables.index ADDED
Binary file (5.77 kB). View file