example-pokemon-classifier/dvc.lock at main · iterative/example-pokemon-classifier · GitHub

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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
schema: '2.0'
stages:
  data_preprocess:
    cmd: python3 src/data_preprocess.py --params params.yaml
    deps:
    - path: data/external/images-gen-1-8
      md5: c48566c030889e71a791e1412d7dddf7.dir
      size: 118179012
      nfiles: 905
    - path: data/external/stats/pokemon-gen-1-8.csv
      md5: b0b029c2dc01bd0e6f2dbf9eeae8c14e
      size: 160616
    - path: src/data_preprocess.py
      md5: f237097fa367335c014cecba4a35a9ba
      size: 3168
    params:
      params.yaml:
        base:
          seed: 42
          pokemon_type_train: Water
        data_preprocess:
          source_directory: data/external
          destination_directory: data/processed
          dataset_labels: stats/pokemon-gen-1-8.csv
          dataset_images: images-gen-1-8
    outs:
    - path: data/processed/pokemon
      md5: c48566c030889e71a791e1412d7dddf7.dir
      size: 118179012
      nfiles: 905
    - path: data/processed/pokemon-with-image-paths.csv
      md5: 7020d2b7ad1b2ee7c2cc3285c7e472ef
      size: 114333
    - path: data/processed/pokemon.csv
      md5: 01120e534796a4006fdfa92c699234b5
      size: 47039
  data_load:
    cmd: python3 src/data_load.py --params params.yaml
    deps:
    - path: data/processed/pokemon
      md5: c48566c030889e71a791e1412d7dddf7.dir
      size: 118179012
      nfiles: 905
    - path: data/processed/pokemon-with-image-paths.csv
      md5: 7020d2b7ad1b2ee7c2cc3285c7e472ef
      size: 114333
    - path: src/data_load.py
      md5: 3beae88ab58ae0cb8d355477e04f3574
      size: 2073
    params:
      params.yaml:
        base:
          seed: 42
          pokemon_type_train: Water
        data_preprocess:
          source_directory: data/external
          destination_directory: data/processed
          dataset_labels: stats/pokemon-gen-1-8.csv
          dataset_images: images-gen-1-8
    outs:
    - path: data/processed/X.pckl
      md5: 8f6555c6159e27739fd97a61ace395a4
      size: 2891610169
    - path: data/processed/X_test.pckl
      md5: 23c3aa551e6826615b71e592972997a8
      size: 581210168
    - path: data/processed/X_train.pckl
      md5: 364b2c4e8fe73f88a79ef5da6416abac
      size: 2310400169
    - path: data/processed/y.pckl
      md5: f156295a269847ea043fff40f8cca0a7
      size: 7072
    - path: data/processed/y_test.pckl
      md5: 05695e9b10f3c84048b6c6d0af01bd30
      size: 3247
    - path: data/processed/y_train.pckl
      md5: 25b4a8bb684e918396014650bfcfeffd
      size: 10913
  train:
    cmd: python3 src/train.py --params params.yaml
    deps:
    - path: data/processed/X_test.pckl
      md5: 23c3aa551e6826615b71e592972997a8
      size: 581210168
    - path: data/processed/X_train.pckl
      md5: 364b2c4e8fe73f88a79ef5da6416abac
      size: 2310400169
    - path: data/processed/y_test.pckl
      md5: 05695e9b10f3c84048b6c6d0af01bd30
      size: 3247
    - path: data/processed/y_train.pckl
      md5: 25b4a8bb684e918396014650bfcfeffd
      size: 10913
    - path: src/train.py
      md5: bf934d6c1c816835fb84b3e1e121a5e2
      size: 4085
    params:
      params.yaml:
        base:
          seed: 42
          pokemon_type_train: Water
        data_preprocess:
          source_directory: data/external
          destination_directory: data/processed
          dataset_labels: stats/pokemon-gen-1-8.csv
          dataset_images: images-gen-1-8
        train:
          test_size: 0.2
          learning_rate: 0.001
          epochs: 15
          batch_size: 120
    outs:
    - path: outputs/model
      md5: bb9347bd95f57b0024b051a77728fd23.dir
      size: 1507711
      nfiles: 4
    - path: outputs/train_history.png
      md5: 8ad98617c7bca9b3a5adfc754a553365
      size: 50476
  evaluate:
    cmd: python3 src/evaluate.py --params params.yaml
    deps:
    - path: outputs/model/keras_metadata.pb
      md5: c7f12fbd11650bd6ea7c3fdd7b53e8f4
      size: 17311
    - path: outputs/model/saved_model.pb
      md5: de5ed7df9f7e78383e03bbd06e6ff882
      size: 194467
    - path: outputs/train_history.png
      md5: 8ad98617c7bca9b3a5adfc754a553365
      size: 50476
    params:
      params.yaml:
        base:
          seed: 42
          pokemon_type_train: Water
    outs:
    - path: outputs/confusion_matrix.png
      md5: 28be5f11bd7bd99093f525b704d08ea4
      size: 22184
    - path: outputs/metrics.yaml
      md5: f102b8baf582402c9966864bc6f1684e
      size: 103