-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.yaml
36 lines (32 loc) · 904 Bytes
/
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
stages:
data_load:
cmd: python src/data_load/data_load.py --test_size=0.35
deps:
- src/data_load/data_load.py
outs:
- data/raw/lenta.csv
- data/raw/train.csv
- data/raw/test.csv
build_train:
cmd: python src/feature/build_feature.py --mode=train --sample_size=10000
deps:
- src/feature/build_feature.py
- data/raw/train.csv
outs:
- data/processed/train_dataset.pkl
build_test:
cmd: python src/feature/build_feature.py --mode=test --sample_size=1000
deps:
- src/feature/build_feature.py
- data/raw/test.csv
outs:
- data/processed/test_dataset.pkl
train_model:
cmd: python src/model/train_model.py --epoch=1
deps:
- src/model/train_model.py
- src/model/model.py
- data/processed/train_dataset.pkl
- data/processed/test_dataset.pkl
outs:
- models/model.torch