libokj's picture
Upload 154 files
416833a
raw
history blame
365 Bytes
_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}