-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.yaml
42 lines (42 loc) · 1.23 KB
/
dvc.yaml
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:
linting:
cmd: poetry run mllint --output reports/mllint_report.md -f
deps:
- .pylintrc
- .mllint.yml
load_data:
cmd: python src/load_data.py
deps:
- src/load_data.py
outs:
- data/external/a1_RestaurantReviews_HistoricDump.tsv
- data/external/a2_RestaurantReviews_FreshDump.tsv
preprocessing:
cmd: python src/preprocessing.py
deps:
- src/preprocessing.py
- data/external/a1_RestaurantReviews_HistoricDump.tsv
outs:
- data/processed/corpus.joblib
training:
cmd: python src/training.py
deps:
- src/training.py
- data/external/a1_RestaurantReviews_HistoricDump.tsv
- data/processed/corpus.joblib
outs:
- data/models/c1_BoW_Sentiment_Model.pkl
- data/models/c2_Classifier_Sentiment_Model
evaluation:
cmd: python src/evaluation.py --output reports/model_evaluation.json
deps:
- src/evaluation.py
- data/models/c2_Classifier_Sentiment_Model
metrics:
- reports/model_evaluation.json
testing:
cmd: pytest --junitxml=reports/tests-report.xml --cov=src --cov-report=xml:reports/coverage-report.xml
deps:
- src/evaluation.py
- data/models/c1_BoW_Sentiment_Model.pkl
- data/models/c2_Classifier_Sentiment_Model