forked from feldlime/RecoServiceTemplate
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.yml
25 lines (21 loc) · 774 Bytes
/
config.yml
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
data:
interactions_path: data/kion_train/interactions.csv
items_path: data/kion_train/items.csv
users_path: data/kion_train/users.csv
user_knn:
model_path: models/userknn_bm25.dill
users_mapping_path: models/users_mapping.dill
users_inv_mapping_path: models/users_inv_mapping.dill
n_users: 30
most_popular:
model_path: models/most_popular.pkl
lightfm:
model_path: models/lightfm.dill
users_mapping_path: models/users_mapping.dill
items_inv_mapping_path: models/items_inv_mapping.dill
hybrid:
ranker: models/lgbm_ranker_model_full.dill
popular_candidates: models/candidates/popular_candidates.csv
als_candidates: models/candidates/als_candidates.csv
lfm_candidates: models/candidates/lfm_candidates.csv
ranker_data: models/ranker_data.csv