emo-classifier/dvc.yaml at main · stdiff/emo-classifier · GitHub

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
stages:
  ingestion:
    cmd: poetry run ingestion
    outs:
      - data/train.parquet
      - data/dev.parquet
      - data/test.parquet
  train:
    cmd: poetry run train
    deps:
      - data/train.parquet
    outs:
      - emo_classifier/artifact/
    metrics:
      - emo_classifier/resources/TrainingMetrics.json:
          cache: false
  evaluate:
    cmd: poetry run jupyter nbconvert --to HTML --execute notebook/error_analysis.ipynb
    deps:
      - data/dev.parquet
      - data/test.parquet
      - emo_classifier/artifact
    outs:
      - emo_classifier/resources/dev_predictions.parquet
      - emo_classifier/resources/Thresholds.json:
          cache: false
      - notebook/error_analysis.html
    metrics:
      - emo_classifier/resources/DevMetrics.json:
          cache: false
      - emo_classifier/resources/TestMetrics.json:
          cache: false