-
Notifications
You must be signed in to change notification settings - Fork 3
/
dvc.lock
35 lines (35 loc) · 974 Bytes
/
dvc.lock
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
schema: '2.0'
stages:
preprocessing:
cmd: python src/preprocessing.py --path=params.yaml
deps:
- path: src/preprocessing.py
md5: b8506cf21013083d5a73e16c3fe5c42b
size: 3570
params:
params.yaml:
base:
batch_size: 32
paths:
data_root: data
zip_data: gender-classification-dataset.zip
train_path: Training
val_path: Validation
processed_train_path: train_image_batches.pk
processed_val_path: val_image_batches.pk
model: model/GR.h5
preprocessing:
x_col: Filepath
y_col: Label
target_size: 224
color_mode: rgb
class_mode: categorical
shuffle: true
seed: 0
rotation_range: 30
zoom_range: 0.15
width_shift_range: 0.2
height_shift_range: 0.2
shear_range: 0.15
horizontal_flip: true
fill_mode: nearest