ium_464953/dvc.yaml
Michal Gulczynski 5eb5fb7172 added dvc.yaml
2024-05-26 20:58:02 +02:00

21 lines
418 B
YAML

stages:
train:
cmd: python model_creator_dvc.py 0.2 100
deps:
- model_creator_dvc.py
- spotify_songs.csv
- Spotify_Dataset.csv
outs:
- model.pkl
- docker_test_dataset.csv
predict:
cmd: python use_model_dvc.py 1
deps:
- use_model_dvc.py
- model.pkl
- docker_test_dataset.csv
outs:
- spotify_genre_predictions.csv
- metrics_df.csv