number-ocr-machine-learning/dvc.lock at main · Sailing-Ranking/number-ocr-machine-learning · 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
schema: '2.0'
stages:
  train:
    cmd: pipenv run python src/02_train.py
    params:
      params.yaml:
        compile:
          metrics: accuracy
        model:
          filters: 32
          kernel_initializer: he_uniform
          input_shape:
          - 28
          - 28
          - 1
          activation: relu
          out_activation: softmax
          pooling:
          - 2
          - 2
          kernel_size:
          - 3
          - 3
          dropout: 0.5
        train:
          validation_split: 0.25
          epochs: 3
    outs:
    - path: data/experiment/cnn_classifier
      md5: c53c937072a7e6e6652c157fcecb1baa.dir
      size: 1103955
      nfiles: 5
    - path: data/experiment/history.json
      md5: 0cd82ce327d8bd7fb26255fc62f115a4
      size: 217
  visualize:
    cmd: pipenv run python src/04_visualize.py
    deps:
    - path: data/experiment/history.json
      md5: 0cd82ce327d8bd7fb26255fc62f115a4
      size: 217
    - path: data/experiment/metrics.json
      md5: 8821d7b04d8cb41f3448a9ce9c3ee107
      size: 114
    - path: data/experiment/predictions.csv
      md5: 3fb6628bea3b4f6b455d43fa5fd187bf
      size: 88905
    outs:
    - path: data/experiment/plots/acc_vs_val_acc.png
      md5: 42267ff5dd07e0405864982324a7d885
      size: 36551
    - path: data/experiment/plots/confusion_matrix.png
      md5: 8914602297a258de2cef0335e6bed466
      size: 85499
    - path: data/experiment/plots/loss_vs_val_loss.png
      md5: 47fc039fb5733a590bf4ce043c652325
      size: 38879
  evaluate:
    cmd: pipenv run python src/03_evaluate.py
    deps:
    - path: data/experiment/cnn_classifier
      md5: c53c937072a7e6e6652c157fcecb1baa.dir
      size: 1103955
      nfiles: 5
    outs:
    - path: data/experiment/metrics.json
      md5: 8821d7b04d8cb41f3448a9ce9c3ee107
      size: 114
    - path: data/experiment/predictions.csv
      md5: 3fb6628bea3b4f6b455d43fa5fd187bf
      size: 88905
  setup:
    cmd: pipenv run python src/01_setup.py
  test:
    cmd: pipenv run test src/05_test.py
    deps:
    - path: data/experiment/metrics.json
      md5: 8821d7b04d8cb41f3448a9ce9c3ee107
      size: 114