21 lines
418 B
YAML
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
|