forked from SylviaWhittle/catsnet-dvc
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.lock
102 lines (102 loc) · 2.46 KB
/
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
schema: '2.0'
stages:
data_split:
cmd: python src/data_split.py
deps:
- path: data/all_data/
hash: md5
md5: 15f848fc495bd9d650462e65288f286c.dir
size: 157899315
nfiles: 317
- path: src/data_split.py
hash: md5
md5: 8d802fe12f83502b7c7c0c012e469f4f
size: 3465
params:
params.yaml:
base:
random_seed: 0
model_image_size: 256
data_split:
test_split: 0.2
data_dir: data/all_data/
train_data_dir: data/train/
test_data_dir: data/test/
outs:
- path: data/test
hash: md5
md5: e8766d24ccc24dcc2c3d7c2411524b01.dir
size: 28969711
nfiles: 64
- path: data/train
hash: md5
md5: b863265b49281c883ce49f3d3365369a.dir
size: 128923456
nfiles: 252
train:
cmd: python src/train.py
deps:
- path: data/train/
hash: md5
md5: b863265b49281c883ce49f3d3365369a.dir
size: 128923456
nfiles: 252
- path: src/train.py
hash: md5
md5: e33e12d02baa85f3e9f9919bc2efa2c5
size: 8209
params:
params.yaml:
base:
random_seed: 0
model_image_size: 256
train:
train_data_dir: data/train/
model_save_dir: models/
batch_size: 5
epochs: 5
activation_function: sigmoid
learning_rate: 0.0001
norm_upper_bound: 7
norm_lower_bound: -1
validation_split: 0.2
outs:
- path: models/catsnet_model.keras
hash: md5
md5: 36e676188d71b155f0db7c5bf08a9464
size: 23455421
- path: results/train/
hash: md5
md5: 04a84f286738dfff0049d8ac7b4dbb43.dir
size: 678
nfiles: 5
evaluate:
cmd: python src/evaluate.py
deps:
- path: data/test/
hash: md5
md5: e8766d24ccc24dcc2c3d7c2411524b01.dir
size: 28969711
nfiles: 64
- path: models/catsnet_model.keras
hash: md5
md5: 36e676188d71b155f0db7c5bf08a9464
size: 23455421
- path: src/evaluate.py
hash: md5
md5: 925f98789798e9920bf396e614d4b29b
size: 5764
params:
params.yaml:
base:
random_seed: 0
model_image_size: 256
evaluate:
model_path: models/catsnet_model.keras
test_data_dir: data/test/
outs:
- path: results/evaluate/
hash: md5
md5: 808749d6794fb63a556b938ca2572a73.dir
size: 3182997
nfiles: 17