Hugging Face's logo

Spaces:
libokj
/
DeepSEQreen_fast_build
Running on CPU Upgrade

App Files Community
DeepSEQreen_fast_build
File size: 711 Bytes 
416833a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# @package _global_
model:
  predictor:
    _target_: deepscreen.models.predictors.bacpi.BACPI

    n_atom: 20480
    n_amino: 8448
    comp_dim: 80
    prot_dim: 80
    latent_dim: 80
    gat_dim: 50
    num_head: 3
    dropout: 0.1
    alpha: 0.1
    window: 5
    layer_cnn: 3
  optimizer:
    lr: 5e-4

data:
  batch_size: 16

  collator:
    automatic_padding: True

  drug_featurizer:
    _target_: deepscreen.models.predictors.bacpi.drug_featurizer
    _partial_: true
    radius: 2

  protein_featurizer:
    _target_: deepscreen.models.predictors.bacpi.split_sequence
    _partial_: true
    ngram: 3
#  collator:
#    _target_: deepscreen.models.predictors.transformer_cpi_2.pack
#    _partial_: true