configs/preset/deep_conv_dti.yaml

# @package _global_
defaults:
  - override /data/drug_featurizer: ecfp
  - override /data/protein_featurizer: label

model:
  predictor:
    _target_: deepscreen.models.predictors.deep_conv_dti.DeepConvDTI

    activation:
      _target_: torch.nn.ELU

    dropout: 0.0
    drug_layers: [512, 128]
    protein_windows: [10, 15, 20, 25, 30]
    n_filters: 128
    decay: 0.0001
    convolution: true
    protein_layers: [128,]
    fc_layers: [128,]

data:
  batch_size: 512