|
@@ -1,7 +1,6 @@
|
|
|
package com.tzld.piaoquan.recommend.server.service.rank.strategy;
|
|
|
|
|
|
import com.ctrip.framework.apollo.spring.annotation.ApolloJsonValue;
|
|
|
-import com.google.common.reflect.TypeToken;
|
|
|
import com.tzld.piaoquan.recommend.server.common.base.RankItem;
|
|
|
import com.tzld.piaoquan.recommend.server.model.Video;
|
|
|
import com.tzld.piaoquan.recommend.server.service.FeatureService;
|
|
@@ -10,7 +9,6 @@ import com.tzld.piaoquan.recommend.server.service.rank.extractor.ExtractorUtils;
|
|
|
import com.tzld.piaoquan.recommend.server.service.recall.strategy.*;
|
|
|
import com.tzld.piaoquan.recommend.server.service.score.ScorerUtils;
|
|
|
import com.tzld.piaoquan.recommend.server.util.CommonCollectionUtils;
|
|
|
-import com.tzld.piaoquan.recommend.server.util.JSONUtils;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.commons.collections4.MapUtils;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
@@ -35,7 +33,6 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
Map<String, double[]> bucketsMap = new HashMap<>();
|
|
|
Map<String, Double> bucketsLen = new HashMap<>();
|
|
|
|
|
|
- String CLASS_NAME = this.getClass().getSimpleName();
|
|
|
@Override
|
|
|
public List<Video> mergeAndRankRovRecall(RankParam param) {
|
|
|
Map<String, Double> mergeWeight = this.mergeWeight != null ? this.mergeWeight : new HashMap<>(0);
|
|
@@ -72,13 +69,12 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
rovRecallRank.addAll(v1);
|
|
|
setVideo.addAll(v1.stream().map(Video::getVideoId).collect(Collectors.toSet()));
|
|
|
|
|
|
-
|
|
|
//-------------------排-------------------
|
|
|
//-------------------序-------------------
|
|
|
//-------------------逻-------------------
|
|
|
//-------------------辑-------------------
|
|
|
|
|
|
- // TODO 1 批量获取特征 省份参数要对齐 headvid 要传递过来!
|
|
|
+ // 1 批量获取特征 省份参数要对齐 headvid 要传递过来!
|
|
|
List<String> vids = CommonCollectionUtils.toListDistinct(rovRecallRank, v -> String.valueOf(v.getVideoId()));
|
|
|
|
|
|
// k1:视频、k2:表、k3:特征、v:特征值
|
|
@@ -90,7 +86,7 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
Map<String, Map<String, Map<String, String>>> featureOriginVideo = feature.getVideoFeature();
|
|
|
|
|
|
|
|
|
- // TODO 2 特征处理
|
|
|
+ // 2 特征处理
|
|
|
Map<String, Double> userFeatureMapDouble = new HashMap<>();
|
|
|
String mid = param.getMid();
|
|
|
Map<String, String> c1 = featureOriginUser.getOrDefault("alg_mid_feature_play", new HashMap<>());
|
|
@@ -98,8 +94,8 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
Map<String, String> c3 = featureOriginUser.getOrDefault("alg_mid_feature_play_tags", new HashMap<>());
|
|
|
Map<String, String> c4 = featureOriginUser.getOrDefault("alg_mid_feature_return_tags", new HashMap<>());
|
|
|
Map<String, String> c5 = featureOriginUser.getOrDefault("alg_mid_feature_share_tags", new HashMap<>());
|
|
|
- Map<String, String> c6 = featureOriginUser.getOrDefault("alg_mid_feature_feed_exp_share_tags", new HashMap<>());
|
|
|
- Map<String, String> c7 = featureOriginUser.getOrDefault("alg_mid_feature_feed_exp_return_tags", new HashMap<>());
|
|
|
+ Map<String, String> c6 = featureOriginUser.getOrDefault("alg_mid_feature_feed_exp_share_tags_v2", new HashMap<>());
|
|
|
+ Map<String, String> c7 = featureOriginUser.getOrDefault("alg_mid_feature_feed_exp_return_tags_v2", new HashMap<>());
|
|
|
Map<String, String> c8 = featureOriginUser.getOrDefault("alg_mid_feature_sharecf", new HashMap<>());
|
|
|
Map<String, String> c9 = featureOriginUser.getOrDefault("alg_mid_feature_returncf", new HashMap<>());
|
|
|
|
|
@@ -169,21 +165,21 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
for (RankItem item : rankItems) {
|
|
|
Map<String, Double> featureMap = new HashMap<>();
|
|
|
String vid = item.getVideoId() + "";
|
|
|
- Map<String, String> b1 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_all_exp", new HashMap<>());
|
|
|
+ Map<String, String> b1 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_all_exp_v2", new HashMap<>());
|
|
|
Map<String, String> b2 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_all_share", new HashMap<>());
|
|
|
Map<String, String> b3 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_all_return", new HashMap<>());
|
|
|
- Map<String, String> b6 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_exp2share", new HashMap<>());
|
|
|
+ Map<String, String> b6 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_exp2share_v2", new HashMap<>());
|
|
|
Map<String, String> b7 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_share2return", new HashMap<>());
|
|
|
|
|
|
- Map<String, String> b8 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_exp", new HashMap<>());
|
|
|
- Map<String, String> b9 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_root_share", new HashMap<>());
|
|
|
- Map<String, String> b10 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_root_return", new HashMap<>());
|
|
|
- Map<String, String> b11 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_exp", new HashMap<>());
|
|
|
- Map<String, String> b12 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_root_share", new HashMap<>());
|
|
|
- Map<String, String> b13 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_root_return", new HashMap<>());
|
|
|
- Map<String, String> b17 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_exp", new HashMap<>());
|
|
|
- Map<String, String> b18 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_root_share", new HashMap<>());
|
|
|
- Map<String, String> b19 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_root_return", new HashMap<>());
|
|
|
+ Map<String, String> b8 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_exp_v2", new HashMap<>());
|
|
|
+ Map<String, String> b9 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_root_share_v2", new HashMap<>());
|
|
|
+ Map<String, String> b10 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_noflow_root_return_v2", new HashMap<>());
|
|
|
+ Map<String, String> b11 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_exp_v2", new HashMap<>());
|
|
|
+ Map<String, String> b12 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_root_share_v2", new HashMap<>());
|
|
|
+ Map<String, String> b13 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_flow_root_return_v2", new HashMap<>());
|
|
|
+ Map<String, String> b17 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_exp_v2", new HashMap<>());
|
|
|
+ Map<String, String> b18 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_root_share_v2", new HashMap<>());
|
|
|
+ Map<String, String> b19 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_feed_province_root_return_v2", new HashMap<>());
|
|
|
|
|
|
List<Tuple4> originData = Arrays.asList(
|
|
|
new Tuple4(b1, b2, b3, "b123"),
|
|
@@ -254,7 +250,7 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
- Map<String, String> d1 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_recsys_feature_cf_i2i_new", new HashMap<>());
|
|
|
+ Map<String, String> d1 = featureOriginVideo.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_recsys_feature_cf_i2i_new_v2", new HashMap<>());
|
|
|
if (!d1.isEmpty()) {
|
|
|
featureMap.put("d1_exp", Double.parseDouble(d1.getOrDefault("exp", "0")));
|
|
|
featureMap.put("d1_return_n", Double.parseDouble(d1.getOrDefault("return_n", "0")));
|
|
@@ -296,66 +292,37 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
item.featureMap = featureMap;
|
|
|
}
|
|
|
|
|
|
- // 排序模型
|
|
|
+ // 3 排序
|
|
|
Map<String, String> sceneFeatureMap = new HashMap<>(0);
|
|
|
- List<RankItem> items = ScorerUtils.getScorerPipeline("feeds_score_config_20240609.conf")
|
|
|
- .scoring(sceneFeatureMap, userFeatureMap, rankItems);
|
|
|
|
|
|
- // 排序融合分
|
|
|
- String redisScoreKey = "redis:vid_hasreturn1d7d_rov:";
|
|
|
- Map<String, Map<String, String>> vid2MapFeature = this.getVideoRedisFeature(vids, redisScoreKey);
|
|
|
+ List<RankItem> items = ScorerUtils.getScorerPipeline("feeds_score_config_xgb_20240828.conf")
|
|
|
+ .scoring(sceneFeatureMap, userFeatureMap, rankItems);
|
|
|
+ Map<String, Map<String, String>> vid2MapFeature = this.getVideoRedisFeature(vids, "redis:vid_hasreturn_rov:");
|
|
|
+ Map<String, Map<String, String>> vid2VovFeatureMap = this.getVideoRedisFeature(vids, "redis:vid_vov_1d3d:");
|
|
|
+ double alpha_vov = mergeWeight.getOrDefault("alpha_vov", 2.0);
|
|
|
List<Video> result = new ArrayList<>();
|
|
|
- Double chooseFunction = mergeWeight.getOrDefault("chooseFunction", 0.0);
|
|
|
- Double rosDefault = mergeWeight.getOrDefault("rosDefault", 0.1);
|
|
|
- Double has_1d = mergeWeight.getOrDefault("has_1d", 1.0);
|
|
|
- Double has_2d = mergeWeight.getOrDefault("has_2d", 1.0);
|
|
|
- Double has_3d = mergeWeight.getOrDefault("has_3d", 1.0);
|
|
|
- Double has_4d = mergeWeight.getOrDefault("has_4d", 0.0);
|
|
|
- Double has_5d = mergeWeight.getOrDefault("has_5d", 0.0);
|
|
|
- Double has_6d = mergeWeight.getOrDefault("has_6d", 0.0);
|
|
|
- Double has_7d = mergeWeight.getOrDefault("has_7d", 0.0);
|
|
|
+// String hasReturnRovKey = mergeWeight.getOrDefault("hasReturnRovKey", 1.0) < 0.5 ? "rate_1" : "rate_n";
|
|
|
+// Double chooseFunction = mergeWeight.getOrDefault("chooseFunction", 0.0);
|
|
|
|
|
|
for (RankItem item : items) {
|
|
|
double score = 0.0;
|
|
|
- Map<String, String> hasReturnRovScoreMap = vid2MapFeature.getOrDefault(item.getVideoId() + "", new HashMap<>());
|
|
|
- double hasReturnRovScore_1d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_1d", "0"));
|
|
|
- double hasReturnRovScore_2d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_2d", "0"));
|
|
|
- double hasReturnRovScore_3d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_3d", "0"));
|
|
|
- double hasReturnRovScore_4d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_4d", "0"));
|
|
|
- double hasReturnRovScore_5d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_5d", "0"));
|
|
|
- double hasReturnRovScore_6d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_6d", "0"));
|
|
|
- double hasReturnRovScore_7d = Double.parseDouble(hasReturnRovScoreMap.getOrDefault("rate_n_7d", "0"));
|
|
|
- double hasReturnRovScore = (
|
|
|
- hasReturnRovScore_1d + hasReturnRovScore_2d + hasReturnRovScore_3d +
|
|
|
- hasReturnRovScore_4d + hasReturnRovScore_5d + hasReturnRovScore_6d
|
|
|
- + hasReturnRovScore_7d
|
|
|
- ) / 7;
|
|
|
- if (has_1d + has_2d + has_3d + has_4d + has_5d + has_6d + has_7d > 0){
|
|
|
- hasReturnRovScore = (
|
|
|
- hasReturnRovScore_1d*has_1d + hasReturnRovScore_2d*has_2d + hasReturnRovScore_3d*has_3d +
|
|
|
- hasReturnRovScore_4d*has_4d + hasReturnRovScore_5d*has_5d + hasReturnRovScore_6d*has_6d
|
|
|
- + hasReturnRovScore_7d*has_7d
|
|
|
- ) / (
|
|
|
- has_1d + has_2d + has_3d + has_4d + has_5d + has_6d + has_7d
|
|
|
- );
|
|
|
- }
|
|
|
- item.getScoresMap().put("hasReturnRovScore_1d", hasReturnRovScore_1d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_2d", hasReturnRovScore_2d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_3d", hasReturnRovScore_3d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_4d", hasReturnRovScore_4d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_5d", hasReturnRovScore_5d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_6d", hasReturnRovScore_6d);
|
|
|
- item.getScoresMap().put("hasReturnRovScore_7d", hasReturnRovScore_7d);
|
|
|
+ double recommend_rate_1d = Double.parseDouble(vid2VovFeatureMap.getOrDefault(item.getVideoId() + "", new HashMap<>())
|
|
|
+ .getOrDefault("recommend_rate_1d", "0"));
|
|
|
+ double recommend_exp_per_1d = Double.parseDouble(vid2VovFeatureMap.getOrDefault(item.getVideoId() + "", new HashMap<>())
|
|
|
+ .getOrDefault("recommend_exp_per_1d", "0"));
|
|
|
+ double vorScore = recommend_rate_1d * recommend_exp_per_1d;
|
|
|
+ item.getScoresMap().put("recommend_rate_1d", recommend_rate_1d);
|
|
|
+ item.getScoresMap().put("recommend_exp_per_1d", recommend_exp_per_1d);
|
|
|
+ item.getScoresMap().put("vorScore", vorScore);
|
|
|
+ item.getScoresMap().put("alpha_vov", alpha_vov);
|
|
|
+ double hasReturnRovScore = Double.parseDouble(vid2MapFeature.getOrDefault(item.getVideoId() + "", new HashMap<>())
|
|
|
+ .getOrDefault("rate_n", "0"));
|
|
|
item.getScoresMap().put("hasReturnRovScore", hasReturnRovScore);
|
|
|
- double fmRov = item.getScoreRov();
|
|
|
- item.getScoresMap().put("fmRov", fmRov);
|
|
|
- if (chooseFunction == 0){
|
|
|
- score = fmRov * (rosDefault + hasReturnRovScore);
|
|
|
- }else if (chooseFunction == 1){
|
|
|
- score = fmRov * (1 + Math.log(hasReturnRovScore + 1));
|
|
|
- }else {
|
|
|
- score = fmRov * (1 + hasReturnRovScore);
|
|
|
- }
|
|
|
+ double rovRateOrigin = item.getScoreRov();
|
|
|
+ item.getScoresMap().put("rovRateOrigin", rovRateOrigin);
|
|
|
+ double rovRate = restoreScore(rovRateOrigin);
|
|
|
+ item.getScoresMap().put("rovRate", rovRate);
|
|
|
+ score = rovRate * (1 + hasReturnRovScore) * (1.0 + alpha_vov * recommend_rate_1d);
|
|
|
|
|
|
Video video = item.getVideo();
|
|
|
video.setScore(score);
|
|
@@ -378,23 +345,8 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
return result;
|
|
|
}
|
|
|
|
|
|
- private Map<String, Map<String, String>> extractVideoFeature(Map<String, Map<String, Map<String, String>>> featureMap) {
|
|
|
- // TODO
|
|
|
- return null;
|
|
|
- }
|
|
|
-
|
|
|
- private Map<String, String> extractSceneFeature(Map<String, Map<String, Map<String, String>>> featureMap) {
|
|
|
- // TODO
|
|
|
- return null;
|
|
|
- }
|
|
|
-
|
|
|
- private Map<String, String> extractUserFeature(Map<String, Map<String, Map<String, String>>> featureMap) {
|
|
|
- // TODO
|
|
|
- return null;
|
|
|
- }
|
|
|
-
|
|
|
private void readBucketFile() {
|
|
|
- InputStream resourceStream = RankStrategy4RegionMergeModelBasic.class.getClassLoader().getResourceAsStream("20240609_bucket_274.txt");
|
|
|
+ InputStream resourceStream = RankStrategy4RegionMergeModelV563.class.getClassLoader().getResourceAsStream("20240609_bucket_274.txt");
|
|
|
if (resourceStream != null) {
|
|
|
try (BufferedReader reader = new BufferedReader(new InputStreamReader(resourceStream))) {
|
|
|
Map<String, double[]> bucketsMap = new HashMap<>();
|
|
@@ -424,7 +376,7 @@ public class RankStrategy4RegionMergeModelV563 extends RankStrategy4RegionMergeM
|
|
|
} else {
|
|
|
log.error("no bucket file");
|
|
|
}
|
|
|
-
|
|
|
}
|
|
|
|
|
|
+
|
|
|
}
|