Skip to content

Commit b7b51f8

Browse files
committed
feat: add config.yaml file to make sure the pipeline runs smoothly
1 parent b0d8a99 commit b7b51f8

File tree

1 file changed

+26
-0
lines changed

1 file changed

+26
-0
lines changed

config.yaml

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
data:
2+
directory: datasets
3+
task_list: null
4+
distances:
5+
molecule:
6+
enabled: true
7+
featurizer: ecfp
8+
method: euclidean
9+
protein:
10+
enabled: false
11+
featurizer: esm2_t33_650M_UR50D
12+
method: cosine
13+
layer: null
14+
combination:
15+
strategy: weighted_average
16+
weights:
17+
molecule: 0.5
18+
protein: 0.5
19+
output:
20+
directory: output
21+
format: csv
22+
save_features: true
23+
compute:
24+
n_jobs: 8
25+
batch_size: 1000
26+
device: auto

0 commit comments

Comments
 (0)