Browse Source

模型打分

zhangbo 10 months ago
parent
commit
1ba22c3d3e

+ 11 - 2
recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/rank/strategy/RankStrategy4RegionMergeModelV999.java

@@ -302,10 +302,19 @@ public class RankStrategy4RegionMergeModelV999 extends RankStrategy4RegionMergeM
 
 
         // TODO 3 排序
         // TODO 3 排序
         Map<String, String> sceneFeatureMap = new HashMap<>(0);
         Map<String, String> sceneFeatureMap = new HashMap<>(0);
-        List<RankItem> rovRecallScore = ScorerUtils.getScorerPipeline("feeds_score_config_20240609.conf")
+        List<RankItem> items = ScorerUtils.getScorerPipeline("feeds_score_config_20240609.conf")
                 .scoring(sceneFeatureMap, userFeatureMap, rankItems);
                 .scoring(sceneFeatureMap, userFeatureMap, rankItems);
+        List<Video> result = new ArrayList<>();
+        for (RankItem item : items) {
+            double score = item.getScoreRov();
+            Video video = item.getVideo();
+            video.setScore(score);
+            video.setSortScore(score);
+            video.setScoresMap(item.getScoresMap());
+            result.add(video);
+        }
+        result.sort(Comparator.comparingDouble(o -> -o.getSortScore()));
 
 
-        List<Video> result = CommonCollectionUtils.toList(rovRecallScore, RankItem::getVideo);
         return result;
         return result;
     }
     }