File size: 1,390 Bytes
0842de0
7e3c514
 
 
d5a6d18
7e3c514
 
 
f9cfbca
 
 
d5a6d18
f901b8e
f9cfbca
 
 
 
e732798
f9cfbca
 
e732798
f9cfbca
 
e732798
0842de0
 
 
d5a6d18
d2bed97
9988244
0842de0
 
 
 
 
7e3c514
0842de0
 
2f9ef59
0842de0
d5a6d18
d2bed97
0842de0
2f9ef59
19df7c5
c6912f8
19df7c5
ca50e48
 
 
 
 
d5a6d18
 
 
 
 
 
f9cfbca
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
57
58
59
60
stages:
  download_data:
    cmd: python src/data/make_dataset.py
    deps:
      - data_params.yml
      - src/data/make_dataset.py
    outs:
      - data/raw
  process_data:
    cmd: python src/data/process_data.py
    deps:
      - data_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:
      - model_params.yml
      - data/processed/train.csv
      - data/processed/validation.csv
      - src/models/train_model.py
    outs:
      - models:
          persist: true
    metrics:
      - reports/training_metrics.csv:
          cache: false
  eval:
    cmd: python src/models/evaluate_model.py
    deps:
      - model_params.yml
      - data/processed/test.csv
      - models
      - src/models/evaluate_model.py
    metrics:
      - reports/evaluation_metrics.csv:
          cache: false
  visualize:
    cmd: streamlit run src/visualization/visualize.py
    deps:
      - models
      - src/visualization/visualize.py
  push_to_hf_hub:
    cmd: python src/models/hf_upload.py
    deps:
      - model_params.yml
      - src/models/hf_upload.py
      - models