Spaces:
Sleeping
Sleeping
File size: 689 Bytes
8f66d04 | 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 | # BioInteract Default Configuration
model:
drug_encoder:
gnn_type: GINE
num_layers: 3
hidden_dim: 256
dropout: 0.2
edge_dim: 16
num_atom_features: 52
jk: last
target_encoder:
esm2_dim: 640 # esm2_t30_150M_UR50D → 640-dim
projection_dim: 256
domain_embed_dim: 32
num_domain_types: 50
use_domain_features: true
interaction:
num_heads: 8
attn_dim: 256
dropout: 0.1
pooling: gated
predictor:
hidden_dims: [256, 128]
dropout: 0.3
task: classification
training:
batch_size: 64
lr: 1.0e-4
weight_decay: 1.0e-5
epochs: 100
patience: 15
seed: 42
|