puneethp
/
RPPP
forked from Dean/RPPP


  
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

	
            stages:
  pre_process:
    cmd: python3 src/pre_process.py
    deps:
      - src/pre_process.py
      - data/raw/rML-raw-data.csv
    params:
      - pre_process
    outs:
      - data/processed/rML-test.csv
      - data/processed/rML-train.csv
  train:
    cmd: python3 src/train.py
    deps:
      - src/model_def.py
      - data/processed/rML-train.csv
      - src/train.py
    params:
      - train
    outs:
      - models/model.pkl
    metrics:
      - models/metrics/train.yaml:
          cache: false
  eval:
    cmd: python3 src/evaluate.py
    deps:
      - src/evaluate.py
      - models/model.pkl
      - data/processed/rML-test.csv
    params:
      - pre_process
      - train
    metrics:
      - models/metrics/test.yaml:
          cache: false