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