修复推荐系统api
Change-Id: I3d35c2a6dd923c70d0eefdbe81916ec863c77df2
diff --git a/Merge/back_rhj/app/services/recommendation_service.py b/Merge/back_rhj/app/services/recommendation_service.py
index 0547f7b..8982bd3 100644
--- a/Merge/back_rhj/app/services/recommendation_service.py
+++ b/Merge/back_rhj/app/services/recommendation_service.py
@@ -27,7 +27,7 @@
args.data_path = './app/user_post_graph.txt' # 修改为帖子图文件
args.pre_model_path = './app/models/recommend/LightGCN_pretrained.pt'
- self.topk = 2 # 默认推荐数量
+ self.topk = 10 # 默认推荐数量
# 初始化多路召回管理器
self.multi_recall = None
@@ -82,7 +82,7 @@
return intersection / union if union > 0 else 0.0
- def mmr_rerank_with_ads(self, post_ids, scores, theta=0.5, target_size=None):
+ def mmr_rerank_with_ads(self, post_ids, scores, theta=0.7, target_size=None):
"""
使用MMR算法重新排序推荐结果，并在过程中加入广告约束
输入: