anime-recommendation-app/dvc.yaml at main · FrienDotJava/anime-recommendation-app · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
stages:
  data_loading:
    cmd: python anime_recommendation_app/data_loading.py
    deps:
    - anime_recommendation_app/data_loading.py
    outs:
    - data/raw
  data_cleaning:
    cmd: python anime_recommendation_app/data_cleaning.py
    deps:
    - anime_recommendation_app/data_cleaning.py
    - data/raw
    outs:
    - data/interim/anime_preprocessed.csv
    - data/interim/rating_cleaned.csv
  feature_engineering:
    cmd: python anime_recommendation_app/feature_engineering.py
    deps:
    - anime_recommendation_app/feature_engineering.py
    - data/interim/anime_preprocessed.csv
    - data/interim/rating_cleaned.csv
    params:
    - data.min_user_ratings
    - data.min_anime_ratings
    outs:
    - data/interim/merged_data.csv
    - models/artifacts/user_to_user_encoded.json
    - models/artifacts/anime_to_anime_encoded.json
    - models/artifacts/anime_encoded_to_anime.json
    - models/artifacts/genre_to_genre_encoded.json
    - models/artifacts/rating_scale.json
  data_preprocessing:
    cmd: python anime_recommendation_app/data_preprocessing.py
    deps:
    - anime_recommendation_app/data_preprocessing.py
    - data/interim/merged_data.csv
    params:
    - data_preprocessing.valid_size
    outs:
    - data/processed
  model_training:
    cmd: python -m anime_recommendation_app.modeling.model_training
    deps:
    - anime_recommendation_app/modeling/model_training.py
    - data/processed
    params:
    - model.regularization_strength
    - model.initializer
    - model.embedding_size
    - model.learning_rate
    - model.epoch
    - model.batch_size
    - model.optimizer
    outs:
    - models/model.keras
  model_evaluation:
    cmd: python -m anime_recommendation_app.modeling.model_evaluation
    deps:
    - anime_recommendation_app/modeling/model_evaluation.py
    - data/processed/test.csv
    - models/model.keras
    metrics:
    - reports/metrics.json