-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdvc.yaml
42 lines (42 loc) · 1.15 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:
extract_data:
cmd: python app/train/extract_data.py --conf=conf/params.yaml
outs:
- weather_data/raw/weather_daily_data.csv
split_data:
cmd: python app/train/split_data.py --conf=conf/params.yaml
deps:
- app/train/split_data.py
- weather_data/raw/weather_daily_data.csv
outs:
- weather_data/test/test_weather_daily_data.csv
- weather_data/train/train_weather_daily_data.csv
train:
cmd: python app/train/train.py --conf=conf/params.yaml
deps:
- app/train/train.py
- weather_data/train/train_weather_daily_data.csv
- conf/params.yaml
outs:
- bin/model.pkl:
cache: false
evaluate:
cmd: python app/train/evaluate.py --conf conf/params.yaml
deps:
- bin/model.pkl
- conf/params.yaml
- weather_data/test/test_weather_daily_data.csv
- weather_data/train/train_weather_daily_data.csv
plots:
- results/evaluate/true_vs_pred.png:
cache: false
metrics:
- results/evaluate/metrics.json:
cache: false
params:
- weather-forecasting/params.yaml
metrics:
- weather-forecasting/metrics.json
plots:
- weather-forecasting/plots/metrics:
x: step