dvc.yaml

stages:
  preprocess:
    cmd: python ./src/preprocess.py ./data/
    deps:
    - ./src/preprocess.py
    - data/likes.csv
    - data/news.csv
    - data/ratings.csv
    - data/users.csv
  train:
    cmd: python ./src/train.py ./data/
    deps:
    - ./data/news_processed.csv
    - ./data/ratings_processed.csv
    - ./data/users_processed.csv
    - ./src/model.py
    - ./src/train.py
    outs:
    - ./data/embeddings.pt
    - ./data/losses.png
    - ./data/test_losses.csv
    - ./data/train_losses.csv
    metrics:
    - ./data/metrics.json:
        cache: true
  evaluate:
    cmd: python ./src/evaluate.py ./data/ 7170
    deps:
    - ./data/embeddings.pt
    - ./data/news_processed.csv
    - ./data/ratings_processed.csv
    - ./data/users_processed.csv
    - ./src/evaluate.py
    - ./src/model.py