File size: 939 Bytes
660e46e
b545bba
54eb468
c3c5c92
8da6527
c3c137f
0df1302
e64671d
faa9fc7
020d7da
2c283f2
bbbf6dc
8e26452
635d7e1
6fd9a0c
ecebcf1
a3ef390
660e46e
 
 
58c8216
d1a2a09
8181a42
660e46e
 
 
 
 
 
 
 
f7a6846
660e46e
8c2bd06
 
660e46e
3ee8789
660e46e
 
 
5e18b6f
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
stages:
  process_data:
    cmd: python src/data/process_data.py
    deps:
      - params.yml
      - data/raw
      - src/data/process_data.py
    outs:
      - data/processed/test.csv:
          persist: true
          cache: true
      - data/processed/train.csv:
          persist: true
          cache: true
      - data/processed/validation.csv:
          persist: true
          cache: true
  train:
    cmd: python src/models/train_model.py
    deps:
      - params.yml
      - data/processed/train.csv
      - data/processed/validation.csv
      - src/models/train_model.py
    outs:
      - models:
          persist: true
    metrics:
      - reports/training_metrics.txt:
          cache: false
  eval:
    cmd: python src/models/evaluate_model.py
    deps:
      - params.yml
      - data/processed/test.csv
      - models
      - src/models/evaluate_model.py
    metrics:
      - reports/metrics.txt:
          cache: false