configs/experiment/example.yaml

# @package _global_

# to execute this experiment run:
# python train.py experiment=example

defaults:
  - override /data: dti_data
  - override /data/drug_featurizer: onehot
  - override /data/protein_featurizer: onehot
  - override /model: dti_model
  - override /model/protein_encoder: cnn
  - override /model/drug_encoder: cnn
  - override /model/decoder: concat_mlp
  - override /callbacks: default
  - override /trainer: default

# all parameters below will be merged with parameters from default configurations set above
# this allows you to overwrite only specified parameters

tags: ["dti"]

seed: 12345

data:
  data_file: davis.csv
  batch_size: 64

model:
  optimizer:
    lr: 0.0001

trainer:
  min_epochs: 1
  max_epochs: 100
  accelerator: gpu