-
Notifications
You must be signed in to change notification settings - Fork 0
/
dvc.yaml
34 lines (34 loc) · 978 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
stages:
create-dataset:
cmd: >
movie_recommender_system create-dataset
--movie-data-path data/raw/tmdb_5000_movie_dataset/movies.csv
--credits-data-path data/raw/tmdb_5000_movie_dataset/credits.csv
--output-path data/interim/dataset.csv
deps:
- data/raw/tmdb_5000_movie_dataset/movies.csv
- data/raw/tmdb_5000_movie_dataset/credits.csv
outs:
- data/interim/dataset.csv
processing:
cmd: >
movie_recommender_system processing
--data-path data/interim/dataset.csv
--config-path params.yaml
--output-path data/processed/dataset.csv
deps:
- data/interim/dataset.csv
- params.yaml
outs:
- data/processed/dataset.csv
params:
- processing
train:
cmd: >
movie_recommender_system train-model
--data-path data/processed/dataset.csv
--output-path models/model.pkl
deps:
- data/processed/dataset.csv
outs:
- models/model.pkl