test_dvc/params.yaml

12 lines
231 B
YAML

prepare:
data_dir: data/raw
save_dir: data/processed
train_valid_split: [0.7, 0.3]
random_seed: 0
train:
data_dir: data/processed
epochs: 5
batch_size: 256
learning_rate: 0.00005
evaluate:
data_dir: data/processed