olga-gorun
/
mnist_dagshub


  
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

	
            stages:
  load_mnist:
    cmd: PYTHONPATH=../code/ python code/load_mnist_mlp.py
    deps:
    - code/load_mnist_mlp.py
    params:
    - load_mnist
    outs:
    - data/X_test_plain.npy
    - data/X_train_plain.npy
    - data/X_val_plain.npy
    - data/Y_test_plain.npy
    - data/Y_train_plain.npy
    - data/Y_val_plain.npy
  train:
    cmd: PYTHONPATH=../code/ python code/model_train_mlp.py
    deps:
    - code/common.py
    - code/model_train_mlp.py
    - data/X_train_plain.npy
    - data/X_val_plain.npy
    - data/Y_train_plain.npy
    - data/Y_val_plain.npy
    params:
    - train
    outs:
    - data/model/model.h5
    - data/model/model.json
    live:
      training_metrics:
        summary: true
        html: true
  test:
    cmd: PYTHONPATH=../code/ python code/model_test_mlp.py
    deps:
    - code/model_test_mlp.py
    - data/X_test_plain.npy
    - data/Y_test_plain.npy
    - data/model/model.h5
    - data/model/model.json
    metrics:
    - metrics.yaml:
        cache: false