ium_464953/dvc.yaml

21 lines
418 B
YAML
Raw Normal View History

2024-05-26 20:58:02 +02:00
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