|
@@ -72,15 +72,15 @@ public class RankService {
|
|
|
}
|
|
|
|
|
|
List<Video> rovRecallRank = mergeAndRankRovRecall(param);
|
|
|
- log.info("mergeAndRankRovRecall rovRecallRank={}", JSONUtils.toJson(rovRecallRank));
|
|
|
+
|
|
|
List<Video> flowPoolRank = mergeAndRankFlowPoolRecall(param);
|
|
|
- log.info("mergeAndRankFlowPoolRecall flowPoolRank={}", JSONUtils.toJson(flowPoolRank));
|
|
|
+
|
|
|
|
|
|
removeDuplicate(param, rovRecallRank, flowPoolRank);
|
|
|
|
|
|
- log.info("removeDuplicate rovRecallRank={}, flowPoolRank={}",
|
|
|
- JSONUtils.toJson(rovRecallRank),
|
|
|
- JSONUtils.toJson(flowPoolRank));
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
|
|
|
|
|
|
return mergeAndSort(param, rovRecallRank, flowPoolRank);
|
|
@@ -150,8 +150,8 @@ public class RankService {
|
|
|
.map(t -> param.getRankKeyPrefix() + t.getVideoId())
|
|
|
.collect(Collectors.toList());
|
|
|
List<String> videoScores = redisTemplate.opsForValue().multiGet(videoIdKeys);
|
|
|
- log.info("rank mergeAndRankRovRecall videoIdKeys={}, videoScores={}", JSONUtils.toJson(videoIdKeys),
|
|
|
- JSONUtils.toJson(videoScores));
|
|
|
+
|
|
|
+
|
|
|
if (CollectionUtils.isNotEmpty(videoScores)
|
|
|
&& videoScores.size() == rovRecallRank.size()) {
|
|
|
for (int i = 0; i < videoScores.size(); i++) {
|
|
@@ -184,14 +184,14 @@ public class RankService {
|
|
|
|
|
|
private List<Video> mergeAndRankRovRecallNew(RankParam param) {
|
|
|
UserFeature userFeature = featureRemoteService.getUserFeature(param.getMid());
|
|
|
- log.info("userFeature = {}", JSONUtils.toJson(userFeature));
|
|
|
+
|
|
|
|
|
|
List<Video> recallVideos = param.getRecallResult().mergeRecallVideos();
|
|
|
List<RankItem> rankItems = CommonCollectionUtils.toList(recallVideos, RankItem::new);
|
|
|
|
|
|
List<Long> videoIds = CommonCollectionUtils.toListDistinct(recallVideos, Video::getVideoId);
|
|
|
Map<Long, ItemFeature> videoFeatureMap = featureRemoteService.getVideoFeatureMap(videoIds);
|
|
|
- log.info("ItemFeature = {}", JSONUtils.toJson(videoFeatureMap));
|
|
|
+
|
|
|
|
|
|
for (RankItem rankItem : rankItems) {
|
|
|
rankItem.setItemFeature(videoFeatureMap.get(rankItem.getVideoId()));
|
|
@@ -199,10 +199,10 @@ public class RankService {
|
|
|
|
|
|
|
|
|
ScoreParam scoreParam = convert(param);
|
|
|
- log.info("scoreParam {}", JSONUtils.toJson(scoreParam));
|
|
|
+
|
|
|
List<RankItem> rovRecallScore = ScorerUtils.getScorerPipeline(ScorerUtils.BASE_CONF)
|
|
|
.scoring(scoreParam, userFeature, rankItems);
|
|
|
- log.info("mergeAndRankRovRecallNew rovRecallScore={}", JSONUtils.toJson(rovRecallScore));
|
|
|
+
|
|
|
|
|
|
return CommonCollectionUtils.toList(rovRecallScore, i -> {
|
|
|
|
|
@@ -380,5 +380,4 @@ public class RankService {
|
|
|
}
|
|
|
|
|
|
|
|
|
-
|
|
|
}
|