-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathparams.yaml
More file actions
29 lines (29 loc) · 1.04 KB
/
params.yaml
File metadata and controls
29 lines (29 loc) · 1.04 KB
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
data:
raw_anime_path: ./data/raw/anime.csv
raw_rating_path: ./data/raw/rating.csv
preprocessed_anime_path: ./data/interim/anime_preprocessed.csv
cleaned_rating_path: ./data/interim/rating_cleaned.csv
merged_data_path: ./data/interim/merged_data.csv
train_set_path: ./data/processed/train.csv
test_set_path: ./data/processed/test.csv
min_user_ratings: 100
min_anime_ratings: 100
feature_engineering:
user_to_user_encoded_path: ./models/artifacts/user_to_user_encoded.json
anime_to_anime_encoded_path: ./models/artifacts/anime_to_anime_encoded.json
anime_encoded_to_anime_path: ./models/artifacts/anime_encoded_to_anime.json
genre_to_genre_encoded_path: ./models/artifacts/genre_to_genre_encoded.json
rating_scale_path: ./models/artifacts/rating_scale.json
data_preprocessing:
valid_size: 0.8
model:
regularization_strength: 1e-6
initializer: he_normal
embedding_size: 50
learning_rate: 0.0001
epoch: 100
batch_size: 2048
model_path: ./models/model.keras
optimizer: adam
model_evaluation:
metrics_path: reports/metrics.json