File size: 365 Bytes
416833a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
_target_: deepscreen.data.dti_datamodule.DTIdatamodule

defaults:
  - _self_
  - split: null
  - drug_featurizer: null
  - protein_featurizer: null

task: ${task.task}
n_class: ${oc.select:task.task.n_class,null}

data_dir: ${paths.data_dir}
dataset_name: null

batch_size: 16
train_val_test_split: [0.7, 0.1, 0.2]

num_workers: 0
pin_memory: false

train: ${train}