File size: 873 Bytes
660e46e
b545bba
54eb468
c3c5c92
8da6527
c3c137f
0df1302
e64671d
faa9fc7
020d7da
bbbf6dc
8e26452
6fd9a0c
ecebcf1
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
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
      - data/processed/train.csv:
          persist: true
      - data/processed/validation.csv:
          persist: 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