DAGsHub-Official
/
DAGsHub-Tutorial-MNIST


  
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

	
            schema: '2.0'
stages:
  featurization:
    cmd: python code/featurization.py
    deps:
    - path: code/featurization.py
      md5: ddf63559873ad5aa352565bb3b76bf0b
      size: 1472
    - path: data/test_data.csv
      md5: c807df8d6d804ab2647fc15c3d40f543
      size: 18289443
    - path: data/train_data.csv
      md5: c807df8d6d804ab2647fc15c3d40f543
      size: 18289443
    outs:
    - path: data/norm_params.json
      md5: 855b3da318f0dfd5eb36d4765092473a
      size: 138861
    - path: data/processed_test_data.npy
      md5: 48b7c5120556c80e0c2c08602b53a216
      size: 1280128
    - path: data/processed_train_data.npy
      md5: 48b7c5120556c80e0c2c08602b53a216
      size: 1280128
  training:
    cmd: python code/train_model.py
    deps:
    - path: code/train_model.py
      md5: 655c3242c17b3d0213d7ce4d9f78344d
      size: 1652
    - path: data/processed_train_data.npy
      md5: 48b7c5120556c80e0c2c08602b53a216
      size: 1280128
    outs:
    - path: data/model.pkl
      md5: ed938ddb3939b89a5c05c7cdc1887f48
      size: 740490
    - path: metrics/train_metric.json
      md5: 84dd09d6a04f5fb2fee433b936b0a4dd
      size: 36
  eval:
    cmd: python code/eval.py
    deps:
    - path: code/eval.py
      md5: 75c6e9b5ffa5ab5bbf83b7c3e8de188d
      size: 974
    - path: data/model.pkl
      md5: ed938ddb3939b89a5c05c7cdc1887f48
      size: 740490
    - path: data/processed_test_data.npy
      md5: 48b7c5120556c80e0c2c08602b53a216
      size: 1280128
    outs:
    - path: metrics/eval.json
      md5: aef91ad4f382159925ea132a024806e9
      size: 20