|
@@ -1,18 +1,21 @@
|
|
|
package com.tzld.piaoquan.recommend.server.service.rank.strategy;
|
|
|
|
|
|
+import com.alibaba.fastjson.JSON;
|
|
|
import com.ctrip.framework.apollo.spring.annotation.ApolloJsonValue;
|
|
|
import com.tzld.piaoquan.recommend.server.common.ThreadPoolFactory;
|
|
|
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;
|
|
|
import com.tzld.piaoquan.recommend.server.service.rank.RankParam;
|
|
|
+import com.tzld.piaoquan.recommend.server.service.rank.bo.UserSRBO;
|
|
|
+import com.tzld.piaoquan.recommend.server.service.rank.bo.UserShareReturnProfile;
|
|
|
import com.tzld.piaoquan.recommend.server.service.rank.extractor.ExtractorUtils;
|
|
|
+import com.tzld.piaoquan.recommend.server.service.rank.tansform.NORFeature;
|
|
|
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.FeatureBucketUtils;
|
|
|
-import com.tzld.piaoquan.recommend.server.util.FeatureUtils;
|
|
|
-import com.tzld.piaoquan.recommend.server.util.SimilarityUtils;
|
|
|
+import com.tzld.piaoquan.recommend.server.util.JSONUtils;
|
|
|
import lombok.extern.slf4j.Slf4j;
|
|
|
import org.apache.commons.collections4.MapUtils;
|
|
|
import org.apache.commons.math3.util.Pair;
|
|
@@ -33,16 +36,6 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
@Autowired
|
|
|
private FeatureService featureService;
|
|
|
|
|
|
- private static final List<String> shortPeriod = Arrays.asList("1h", "2h", "4h", "6h", "12h", "24h", "7d");
|
|
|
- private static final List<String> middlePeriod = Arrays.asList("14d", "30d");
|
|
|
- private static final List<String> longPeriod = Arrays.asList("7d", "35d", "90d", "365d");
|
|
|
- private static final List<String> cfRosList = Collections.singletonList("rosn");
|
|
|
- private static final List<String> cfRovList = Collections.singletonList("rovn");
|
|
|
- private static final List<String> videoSimAttrs = Arrays.asList("cate1_list", "cate2", "cate2_list",
|
|
|
- "keywords", "style", "theme", "title", "topic", "user_value");
|
|
|
- private static final List<String> videoCateAttr = Arrays.asList(FeatureUtils.cate1Attr, FeatureUtils.cate2Attr, FeatureUtils.festive1Attr);
|
|
|
- private static final double smoothPlus = 5.0;
|
|
|
-
|
|
|
@Override
|
|
|
public List<Video> mergeAndRankRovRecall(RankParam param) {
|
|
|
Map<String, Double> mergeWeight = this.mergeWeight != null ? this.mergeWeight : new HashMap<>(0);
|
|
@@ -50,6 +43,7 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
//-------------------合-------------------
|
|
|
//-------------------逻-------------------
|
|
|
//-------------------辑-------------------
|
|
|
+
|
|
|
long currentMs = System.currentTimeMillis();
|
|
|
List<Video> oldRovs = new ArrayList<>();
|
|
|
oldRovs.addAll(extractAndSort(param, RegionHRecallStrategy.PUSH_FORM));
|
|
@@ -75,7 +69,7 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
//-------------------新地域召回------------------
|
|
|
List<Video> v1 = extractAndSort(param, RegionRealtimeRecallStrategyV1.PUSH_FORM);
|
|
|
v1 = v1.stream().filter(r -> !setVideo.contains(r.getVideoId())).collect(Collectors.toList());
|
|
|
- v1 = v1.subList(0, Math.min(mergeWeight.getOrDefault("v1", 10.0).intValue(), v1.size()));
|
|
|
+ v1 = v1.subList(0, Math.min(mergeWeight.getOrDefault("v1", 5.0).intValue(), v1.size()));
|
|
|
rovRecallRank.addAll(v1);
|
|
|
setVideo.addAll(v1.stream().map(Video::getVideoId).collect(Collectors.toSet()));
|
|
|
//-------------------scene cf rovn------------------
|
|
@@ -100,15 +94,19 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
List<String> vids = CommonCollectionUtils.toListDistinct(rovRecallRank, v -> String.valueOf(v.getVideoId()));
|
|
|
|
|
|
// k1:视频、k2:表、k3:特征、v:特征值
|
|
|
- String provinceCn = param.getProvince().replaceAll("省$", "");
|
|
|
String headVid = String.valueOf(param.getHeadVid());
|
|
|
- String sceneType = String.valueOf(param.getHotSceneType());
|
|
|
Map<String, Map<String, Map<String, String>>> videoBaseInfoMap = featureService.getVideoBaseInfo(headVid, vids);
|
|
|
- FeatureService.Feature feature = featureService.getNewFeature(provinceCn, param.getMid(), sceneType, headVid, videoBaseInfoMap, vids);
|
|
|
+ FeatureService.Feature feature = featureService.getFeatureV3(param, videoBaseInfoMap, vids);
|
|
|
Map<String, Map<String, String>> featureOriginUser = feature.getUserFeature();
|
|
|
Map<String, Map<String, Map<String, String>>> featureOriginVideo = feature.getVideoFeature();
|
|
|
Map<String, String> headVideoInfo = videoBaseInfoMap.getOrDefault(headVid, new HashMap<>()).getOrDefault("alg_vid_feature_basic_info", new HashMap<>());
|
|
|
|
|
|
+ // 用户信息预处理
|
|
|
+ Map<String, Map<String, String[]>> newC7Map = NORFeature.parseUCFScore(featureOriginUser.getOrDefault("alg_mid_feature_sharecf", new HashMap<>()));
|
|
|
+ Map<String, Map<String, String[]>> newC8Map = NORFeature.parseUCFScore(featureOriginUser.getOrDefault("alg_mid_feature_returncf", new HashMap<>()));
|
|
|
+ UserShareReturnProfile userProfile = parseUserProfile(featureOriginUser);
|
|
|
+ Map<String, Map<String, String>> userBehaviorVideoMap = getUserBehaviorVideoMap(userProfile);
|
|
|
+
|
|
|
// 2 特征处理
|
|
|
Map<String, Double> userFeatureMapDouble = new HashMap<>();
|
|
|
String mid = param.getMid();
|
|
@@ -122,9 +120,6 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
Map<String, String> c8 = featureOriginUser.getOrDefault("alg_mid_feature_sharecf", new HashMap<>());
|
|
|
Map<String, String> c9 = featureOriginUser.getOrDefault("alg_mid_feature_returncf", new HashMap<>());
|
|
|
|
|
|
- // time feature
|
|
|
- addTimeFeature(currentMs, userFeatureMapDouble);
|
|
|
-
|
|
|
if (!c1.isEmpty()) {
|
|
|
userFeatureMapDouble.put("playcnt_6h", Double.parseDouble(c1.getOrDefault("playcnt_6h", "0")));
|
|
|
userFeatureMapDouble.put("playcnt_1d", Double.parseDouble(c1.getOrDefault("playcnt_1d", "0")));
|
|
@@ -221,29 +216,23 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
double share = tuple4.second.isEmpty() ? 0 : Double.parseDouble(tuple4.second.getOrDefault("share_pv_" + prefix2, "0.0"));
|
|
|
double returns = tuple4.third.isEmpty() ? 0 : Double.parseDouble(tuple4.third.getOrDefault("return_uv_" + prefix2, "0.0"));
|
|
|
|
|
|
- double f0 = ExtractorUtils.calLog(exp);
|
|
|
double f1 = ExtractorUtils.calDiv(share, exp);
|
|
|
double f2 = ExtractorUtils.calLog(share);
|
|
|
double f3 = ExtractorUtils.calDiv(returns, exp);
|
|
|
double f4 = ExtractorUtils.calLog(returns);
|
|
|
double f5 = f3 * f4;
|
|
|
- double f6 = ExtractorUtils.calDiv(returns, share);
|
|
|
|
|
|
- String key0 = tuple4.name + "_" + prefix2 + "_" + "EXP";
|
|
|
String key1 = tuple4.name + "_" + prefix2 + "_" + "STR";
|
|
|
String key2 = tuple4.name + "_" + prefix2 + "_" + "log(share)";
|
|
|
String key3 = tuple4.name + "_" + prefix2 + "_" + "ROV";
|
|
|
String key4 = tuple4.name + "_" + prefix2 + "_" + "log(return)";
|
|
|
String key5 = tuple4.name + "_" + prefix2 + "_" + "ROV*log(return)";
|
|
|
- String key6 = tuple4.name + "_" + prefix2 + "_" + "ROS";
|
|
|
|
|
|
- featureMap.put(key0, f0);
|
|
|
featureMap.put(key1, f1);
|
|
|
featureMap.put(key2, f2);
|
|
|
featureMap.put(key3, f3);
|
|
|
featureMap.put(key4, f4);
|
|
|
featureMap.put(key5, f5);
|
|
|
- featureMap.put(key6, f6);
|
|
|
}
|
|
|
}
|
|
|
|
|
@@ -301,20 +290,17 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
featureMap.put("d1_return_n", Double.parseDouble(d1.getOrDefault("return_n", "0")));
|
|
|
featureMap.put("d1_rovn", Double.parseDouble(d1.getOrDefault("rovn", "0")));
|
|
|
}
|
|
|
- // ******************** new feature ********************
|
|
|
- addVideoStatFeature(vid, featureOriginVideo, featureMap);
|
|
|
- //addVideoCFFeature(vid, featureOriginVideo, featureMap);
|
|
|
- addVideoSimFeature(headVideoInfo, videoInfo, featureMap);
|
|
|
- addVideoAttrFeature(videoInfo, featureMap);
|
|
|
- addVideoTimeFeature(currentMs, videoInfo, featureMap);
|
|
|
-
|
|
|
item.featureMapDouble = featureMap;
|
|
|
}
|
|
|
|
|
|
+ // get nor feature
|
|
|
+ Map<String, String> norUserFeatureMap = getNorUserFeature(currentMs, headVideoInfo, userProfile, featureOriginUser);
|
|
|
+ batchGetNorVideoFeature(currentMs, userProfile, headVideoInfo, videoBaseInfoMap,
|
|
|
+ newC7Map, newC8Map, featureOriginUser, userBehaviorVideoMap, featureOriginVideo, rankItems);
|
|
|
+
|
|
|
// 3 连续值特征分桶
|
|
|
readBucketFile();
|
|
|
Map<String, String> userFeatureMap = new HashMap<>(userFeatureMapDouble.size());
|
|
|
- Map<String, String> norUserFeatureMap = FeatureBucketUtils.noBucketFeature(userFeatureMapDouble);
|
|
|
for (Map.Entry<String, Double> entry : userFeatureMapDouble.entrySet()) {
|
|
|
String name = entry.getKey();
|
|
|
Double score = entry.getValue();
|
|
@@ -342,19 +328,12 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
}
|
|
|
}
|
|
|
item.featureMap = featureMap;
|
|
|
- item.norFeatureMap = FeatureBucketUtils.noBucketFeature(featureMapDouble);
|
|
|
}
|
|
|
// 4 排序模型计算
|
|
|
- double fmRovLogBase = mergeWeight.getOrDefault("fmRovLogBase", 2.8);
|
|
|
- double xgbNorScaleType = mergeWeight.getOrDefault("xgbNorScaleType", 0.0);
|
|
|
- double xgbNorBias = mergeWeight.getOrDefault("xgbNorBias", -1.6945);
|
|
|
- double xgbNorWeight = mergeWeight.getOrDefault("xgbNorWeight", 1.8968);
|
|
|
- double xgbNorPowerWeight = mergeWeight.getOrDefault("xgbNorPowerWeight", 1.2216);
|
|
|
- double xgbNorPowerExp = mergeWeight.getOrDefault("xgbNorPowerExp", 1.3217);
|
|
|
- double vorLogWeight = mergeWeight.getOrDefault("vorLogWeight", 10.0);
|
|
|
- double vorLogBase = mergeWeight.getOrDefault("vorLogBase", 2.6);
|
|
|
+ double xgbNorPowerWeight = mergeWeight.getOrDefault("xgbNorPowerWeight", 1.22);
|
|
|
+ double xgbNorPowerExp = mergeWeight.getOrDefault("xgbNorPowerExp", 1.24);
|
|
|
Map<String, String> sceneFeatureMap = new HashMap<>(0);
|
|
|
- List<RankItem> items = ScorerUtils.getScorerPipeline("feeds_score_config_fm_xgb_20250218.conf").scoring(sceneFeatureMap, userFeatureMap, norUserFeatureMap, rankItems);
|
|
|
+ List<RankItem> items = ScorerUtils.getScorerPipeline("feeds_score_config_fm_xgb_20250221.conf").scoring(sceneFeatureMap, userFeatureMap, norUserFeatureMap, rankItems);
|
|
|
// 5 排序公式特征
|
|
|
Map<String, Map<String, String>> vid2MapFeature = this.getVideoRedisFeature(vids, "redis:vid_hasreturn_vor:");
|
|
|
List<Video> result = new ArrayList<>();
|
|
@@ -364,15 +343,13 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
item.getScoresMap().put("fmRovOrigin", fmRovOrigin);
|
|
|
double fmRov = restoreScore(fmRovOrigin);
|
|
|
item.getScoresMap().put("fmRov", fmRov);
|
|
|
- double newFmRov = logRov(fmRov, fmRovLogBase);
|
|
|
double hasReturnRovScore = Double.parseDouble(vid2MapFeature.getOrDefault(item.getVideoId() + "", new HashMap<>()).getOrDefault("rov", "0"));
|
|
|
item.getScoresMap().put("hasReturnRovScore", hasReturnRovScore);
|
|
|
double norXGBScore = item.getScoresMap().getOrDefault("NorXGBScore", 0d);
|
|
|
- double newNorXGBScore = norCalibration(xgbNorScaleType, xgbNorBias, xgbNorWeight, xgbNorPowerWeight, xgbNorPowerExp, norXGBScore);
|
|
|
+ double newNorXGBScore = norPowerCalibration(xgbNorPowerWeight, xgbNorPowerExp, norXGBScore);
|
|
|
double vor = Double.parseDouble(vid2MapFeature.getOrDefault(item.getVideoId() + "", new HashMap<>()).getOrDefault("vor", "0"));
|
|
|
- double newVor = logVor(vorLogWeight, vor, vorLogBase);
|
|
|
item.getScoresMap().put("vor", vor);
|
|
|
- score = newFmRov * (0.1 + newNorXGBScore) * (0.1 + newVor);
|
|
|
+ score = fmRov * (0.1 + newNorXGBScore) * (0.1 + vor);
|
|
|
Video video = item.getVideo();
|
|
|
video.setScore(score);
|
|
|
video.setSortScore(score);
|
|
@@ -396,172 +373,126 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
return result;
|
|
|
}
|
|
|
|
|
|
- private Map<String, String> getVideoOneTypeInfo(String vid, String name,
|
|
|
- Map<String, Map<String, Map<String, String>>> videoAllInfoMap) {
|
|
|
- if (null == videoAllInfoMap) {
|
|
|
- return new HashMap<>();
|
|
|
- }
|
|
|
- return videoAllInfoMap.getOrDefault(vid, new HashMap<>()).getOrDefault(name, new HashMap<>());
|
|
|
- }
|
|
|
-
|
|
|
- private double getVideoOneInfo(String name, Map<String, String> infoMap) {
|
|
|
- if (null == infoMap) {
|
|
|
- return 0.0;
|
|
|
- }
|
|
|
- return infoMap.isEmpty() ? 0 : Double.parseDouble(infoMap.getOrDefault(name, "0.0"));
|
|
|
- }
|
|
|
-
|
|
|
- private void addVideoStatFeature(String vid, Map<String, Map<String, Map<String, String>>> videoAllInfoMap,
|
|
|
- Map<String, Double> featureMap) {
|
|
|
- List<Tuple3> vidStatInfo = Arrays.asList(
|
|
|
- new Tuple3("b20", shortPeriod, getVideoOneTypeInfo(vid, "alg_cate2_feature", videoAllInfoMap)),
|
|
|
- new Tuple3("b21", shortPeriod, getVideoOneTypeInfo(vid, "alg_cate1_feature", videoAllInfoMap)),
|
|
|
- new Tuple3("b22", shortPeriod, getVideoOneTypeInfo(vid, "alg_vid_source_feature", videoAllInfoMap)),
|
|
|
- new Tuple3("b28", shortPeriod, getVideoOneTypeInfo(vid, "alg_sence_type_feature", videoAllInfoMap)),
|
|
|
- new Tuple3("b29", shortPeriod, getVideoOneTypeInfo(vid, "alg_videoid_feature", videoAllInfoMap)),
|
|
|
- new Tuple3("b23", middlePeriod, getVideoOneTypeInfo(vid, "alg_cate2_feature_day", videoAllInfoMap)),
|
|
|
- new Tuple3("b24", middlePeriod, getVideoOneTypeInfo(vid, "alg_cate1_feature_day", videoAllInfoMap)),
|
|
|
- new Tuple3("b25", middlePeriod, getVideoOneTypeInfo(vid, "alg_video_source_feature_day", videoAllInfoMap)),
|
|
|
- new Tuple3("b26", longPeriod, getVideoOneTypeInfo(vid, "alg_video_unionid_feature_day", videoAllInfoMap)),
|
|
|
- new Tuple3("b27", longPeriod, getVideoOneTypeInfo(vid, "alg_vid_feature_day", videoAllInfoMap))
|
|
|
- );
|
|
|
- for (Tuple3 tuple3 : vidStatInfo) {
|
|
|
- String infoType = tuple3.first;
|
|
|
- List<String> infoPeriod = tuple3.second;
|
|
|
- Map<String, String> infoMap = tuple3.third;
|
|
|
- for (String period : infoPeriod) {
|
|
|
- double view = getVideoOneInfo("view_" + period, infoMap);
|
|
|
- double share = getVideoOneInfo("share_" + period, infoMap);
|
|
|
- double return_ = getVideoOneInfo("return_" + period, infoMap);
|
|
|
- double view_hasreturn = getVideoOneInfo("view_hasreturn_" + period, infoMap);
|
|
|
- double share_hasreturn = getVideoOneInfo("share_hasreturn_" + period, infoMap);
|
|
|
- double ros = ExtractorUtils.smoothDiv(return_, share, smoothPlus);
|
|
|
- double rov = ExtractorUtils.smoothDiv(return_, view, smoothPlus);
|
|
|
- double r_cnt = ExtractorUtils.smoothDiv(return_, view_hasreturn, smoothPlus);
|
|
|
- double r_rate = ExtractorUtils.smoothDiv(view_hasreturn, view, smoothPlus);
|
|
|
- double r_cnt4s = ExtractorUtils.smoothDiv(return_, share_hasreturn, smoothPlus);
|
|
|
- double str = ExtractorUtils.smoothDiv(share, view, smoothPlus);
|
|
|
- double s2r_rate = ExtractorUtils.smoothDiv(share_hasreturn, share, smoothPlus);
|
|
|
-
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "view", ExtractorUtils.calLog(view));
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "share", ExtractorUtils.calLog(share));
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "return", ExtractorUtils.calLog(return_));
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "view_hasreturn", ExtractorUtils.calLog(view_hasreturn));
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "share_hasreturn", ExtractorUtils.calLog(share_hasreturn));
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "ros", ros);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "rov", rov);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "r_cnt", r_cnt);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "r_rate", r_rate);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "r_cnt4s", r_cnt4s);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "str", str);
|
|
|
- featureMap.put(infoType + "_" + period + "_" + "s2r_rate", s2r_rate);
|
|
|
- }
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- private void addVideoCFFeature(String vid, Map<String, Map<String, Map<String, String>>> videoAllInfoMap,
|
|
|
- Map<String, Double> featureMap) {
|
|
|
- List<Tuple3> vidCFInfo = Arrays.asList(
|
|
|
- new Tuple3("d2", cfRosList, getVideoOneTypeInfo(vid, "alg_recsys_feature_weak_cf_i2i_scene_ros", videoAllInfoMap)),
|
|
|
- new Tuple3("d3", cfRosList, getVideoOneTypeInfo(vid, "alg_recsys_feature_cf_i2i_scene_ros", videoAllInfoMap)),
|
|
|
- new Tuple3("d4", cfRovList, getVideoOneTypeInfo(vid, "alg_recsys_feature_weak_cf_i2i_scene_rov", videoAllInfoMap)),
|
|
|
- new Tuple3("d5", cfRovList, getVideoOneTypeInfo(vid, "alg_recsys_feature_cf_i2i_scene_rov", videoAllInfoMap))
|
|
|
- );
|
|
|
- for (Tuple3 tuple3 : vidCFInfo) {
|
|
|
- String infoType = tuple3.first;
|
|
|
- List<String> valTypeList = tuple3.second;
|
|
|
- Map<String, String> infoMap = tuple3.third;
|
|
|
- if (!infoMap.isEmpty()) {
|
|
|
- for (String valType : valTypeList) {
|
|
|
- double exp = getVideoOneInfo("exp", infoMap);
|
|
|
- double return_n = getVideoOneInfo("return_n", infoMap);
|
|
|
- double value = getVideoOneInfo(valType, infoMap);
|
|
|
-
|
|
|
- featureMap.put(infoType + "_exp", ExtractorUtils.calLog(exp));
|
|
|
- featureMap.put(infoType + "_return_n", ExtractorUtils.calLog(return_n));
|
|
|
- featureMap.put(infoType + "_" + valType, value);
|
|
|
+ private UserShareReturnProfile parseUserProfile(Map<String, Map<String, String>> userOriginInfo) {
|
|
|
+ if (null != userOriginInfo) {
|
|
|
+ Map<String, String> c9 = userOriginInfo.get("alg_recsys_feature_user_share_return_stat");
|
|
|
+ if (null != c9 && !c9.isEmpty()) {
|
|
|
+ String c9Str = JSONUtils.toJson(c9);
|
|
|
+ if (!c9Str.isEmpty()) {
|
|
|
+ try {
|
|
|
+ return JSON.parseObject(c9Str, UserShareReturnProfile.class);
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("parseObject user profile error! value=[{}]", c9Str, e);
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+ return null;
|
|
|
}
|
|
|
|
|
|
- private void addVideoSimFeature(Map<String, String> headInfo, Map<String, String> rankInfo, Map<String, Double> featureMap) {
|
|
|
- if (!headInfo.isEmpty() && !rankInfo.isEmpty()) {
|
|
|
- List<Future<Pair<String, Double>>> futures = new ArrayList<>();
|
|
|
- for (String attr : videoSimAttrs) {
|
|
|
- String headAttr = headInfo.getOrDefault(attr, "");
|
|
|
- String rankAttr = rankInfo.getOrDefault(attr, "");
|
|
|
- if (!"".equals(headAttr) && !"".equals(rankAttr)) {
|
|
|
- String key = "video_sim_" + attr;
|
|
|
- Future<Pair<String, Double>> future = ThreadPoolFactory.defaultPool().submit(() -> {
|
|
|
- double simScore = SimilarityUtils.word2VecSimilarity(headAttr, rankAttr);
|
|
|
- return Pair.create(key, simScore);
|
|
|
- });
|
|
|
- futures.add(future);
|
|
|
- }
|
|
|
- }
|
|
|
- try {
|
|
|
- for (Future<Pair<String, Double>> future : futures) {
|
|
|
- Pair<String, Double> pair = future.get(1000, TimeUnit.MILLISECONDS);
|
|
|
- featureMap.put(pair.getFirst(), pair.getSecond());
|
|
|
+ private Map<String, Map<String, String>> getUserBehaviorVideoMap(UserShareReturnProfile userProfile) {
|
|
|
+ Set<String> vidSet = new HashSet<>();
|
|
|
+ if (null != userProfile) {
|
|
|
+ for (List<UserSRBO> list : Arrays.asList(userProfile.getM_s_s(), userProfile.getM_r_s(), userProfile.getL_s_s(), userProfile.getL_r_s())) {
|
|
|
+ if (null != list) {
|
|
|
+ for (UserSRBO u : list) {
|
|
|
+ if (null != u) {
|
|
|
+ vidSet.add(u.getId() + "");
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
- } catch (Exception e) {
|
|
|
- log.error("video attr similarity error", e);
|
|
|
}
|
|
|
}
|
|
|
- }
|
|
|
|
|
|
- private void addVideoAttrFeature(Map<String, String> rankInfo, Map<String, Double> featureMap) {
|
|
|
- if (!rankInfo.isEmpty()) {
|
|
|
- for (String attr : videoCateAttr) {
|
|
|
- String attrVal = rankInfo.getOrDefault(attr, "");
|
|
|
- int attrId = FeatureUtils.getAttrId(attr, attrVal);
|
|
|
- if (attrId > 0) {
|
|
|
- String key = String.format("%s@%s@%d", "r", attr, attrId);
|
|
|
- featureMap.put(key, 1.0);
|
|
|
+ Map<String, Map<String, String>> historyVideoMap = new HashMap<>();
|
|
|
+ if (!vidSet.isEmpty()) {
|
|
|
+ Map<String, Map<String, Map<String, String>>> videoMap = featureService.getVideoBaseInfo("", new ArrayList<>(vidSet));
|
|
|
+ if (null != videoMap && !videoMap.isEmpty()) {
|
|
|
+ for (Map.Entry<String, Map<String, Map<String, String>>> entry : videoMap.entrySet()) {
|
|
|
+ String vid = entry.getKey();
|
|
|
+ Map<String, Map<String, String>> map = entry.getValue();
|
|
|
+ if (null != map && map.containsKey("alg_vid_feature_basic_info")) {
|
|
|
+ historyVideoMap.put(vid, map.get("alg_vid_feature_basic_info"));
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+ return historyVideoMap;
|
|
|
}
|
|
|
|
|
|
- private void addVideoTimeFeature(long currentMs, Map<String, String> rankInfo, Map<String, Double> featureMap) {
|
|
|
- double createTime = 1D;
|
|
|
- try {
|
|
|
- if (!rankInfo.isEmpty() && rankInfo.containsKey("gmt_create_timestamp")) {
|
|
|
- String createMsStr = rankInfo.get("gmt_create_timestamp");
|
|
|
- long createMs = Long.parseLong(createMsStr);
|
|
|
- createTime = FeatureUtils.getCreateTime(currentMs, createMs);
|
|
|
- }
|
|
|
- } catch (Exception e) {
|
|
|
- log.error("get video crate time error", e);
|
|
|
- }
|
|
|
- featureMap.put("createTime", createTime);
|
|
|
- }
|
|
|
+ private Map<String, String> getNorUserFeature(long currentMs, Map<String, String> headInfo, UserShareReturnProfile userProfile, Map<String, Map<String, String>> userOriginInfo) {
|
|
|
+ Map<String, Double> featMap = new HashMap<>();
|
|
|
+ // context feature
|
|
|
+ NORFeature.getContextFeature(currentMs, featMap);
|
|
|
+
|
|
|
+ // head video feature
|
|
|
+ NORFeature.getVideoBaseFeature("h", currentMs, headInfo, featMap);
|
|
|
|
|
|
- private void addTimeFeature(long currentMs, Map<String, Double> featureMap) {
|
|
|
- Calendar calendar = Calendar.getInstance();
|
|
|
- calendar.setTimeInMillis(currentMs);
|
|
|
+ // user feature
|
|
|
+ NORFeature.getUserFeature(userOriginInfo, featMap);
|
|
|
+ NORFeature.getUserProfileFeature(userProfile, featMap);
|
|
|
|
|
|
- int week = calendar.get(Calendar.DAY_OF_WEEK);
|
|
|
- int hour = calendar.get(Calendar.HOUR_OF_DAY) + 1;
|
|
|
- featureMap.put("week", week * 1.0);
|
|
|
- featureMap.put("hour", hour * 1.0);
|
|
|
+ return FeatureBucketUtils.noBucketFeature(featMap);
|
|
|
}
|
|
|
|
|
|
- private double norCalibration(double scaleType, double polyBias, double polyWeight, double powerWeight, double powerExp, double score) {
|
|
|
- if (scaleType < 1) {
|
|
|
- return norPolyCalibration(polyBias, polyWeight, score);
|
|
|
- } else {
|
|
|
- return norPowerCalibration(powerWeight, powerExp, score);
|
|
|
- }
|
|
|
+ private Map<String, String> getNorVideoFeature(long currentMs, String vid,
|
|
|
+ UserShareReturnProfile userProfile,
|
|
|
+ Map<String, String> headInfo, Map<String, String> rankInfo,
|
|
|
+ Map<String, Map<String, String[]>> c7Map,
|
|
|
+ Map<String, Map<String, String[]>> c8Map,
|
|
|
+ Map<String, Map<String, String>> userOriginInfo,
|
|
|
+ Map<String, Map<String, String>> historyVideoMap,
|
|
|
+ Map<String, Map<String, Map<String, String>>> videoOriginInfo) {
|
|
|
+ Map<String, Double> featMap = new HashMap<>();
|
|
|
+ // user & video feature
|
|
|
+ NORFeature.getUserTagsCrossVideoFeature("c5", rankInfo, userOriginInfo.get("alg_mid_feature_return_tags"), featMap);
|
|
|
+ NORFeature.getUserTagsCrossVideoFeature("c6", rankInfo, userOriginInfo.get("alg_mid_feature_share_tags"), featMap);
|
|
|
+ NORFeature.getUserCFFeature("c7", vid, c7Map, featMap);
|
|
|
+ NORFeature.getUserCFFeature("c8", vid, c8Map, featMap);
|
|
|
+
|
|
|
+ // rank video feature
|
|
|
+ NORFeature.getVideoBaseFeature("r", currentMs, rankInfo, featMap);
|
|
|
+ NORFeature.getVideoFeature(vid, videoOriginInfo, featMap);
|
|
|
+
|
|
|
+ // head&rank cross feature
|
|
|
+ NORFeature.getHeadRankVideoCrossFeature(headInfo, rankInfo, featMap);
|
|
|
+
|
|
|
+ // user profile & rank cross
|
|
|
+ NORFeature.getProfileVideoCrossFeature(currentMs, userProfile, rankInfo, historyVideoMap, featMap);
|
|
|
+
|
|
|
+ return FeatureBucketUtils.noBucketFeature(featMap);
|
|
|
}
|
|
|
|
|
|
- private double norPolyCalibration(double bias, double weight, double score) {
|
|
|
- double newScore = bias + weight * score;
|
|
|
- if (newScore < score) {
|
|
|
- newScore = score;
|
|
|
+ private void batchGetNorVideoFeature(long currentMs,
|
|
|
+ UserShareReturnProfile userProfile,
|
|
|
+ Map<String, String> headInfo,
|
|
|
+ Map<String, Map<String, Map<String, String>>> videoBaseInfoMap,
|
|
|
+ Map<String, Map<String, String[]>> c7Map,
|
|
|
+ Map<String, Map<String, String[]>> c8Map,
|
|
|
+ Map<String, Map<String, String>> userOriginInfo,
|
|
|
+ Map<String, Map<String, String>> historyVideoMap,
|
|
|
+ Map<String, Map<String, Map<String, String>>> videoOriginInfo,
|
|
|
+ List<RankItem> rankItems) {
|
|
|
+ if (null != rankItems && !rankItems.isEmpty()) {
|
|
|
+ List<Future<Integer>> futures = new ArrayList<>();
|
|
|
+ for (RankItem item : rankItems) {
|
|
|
+ String vid = item.getVideoId() + "";
|
|
|
+ Map<String, String> rankInfo = videoBaseInfoMap.getOrDefault(vid, new HashMap<>()).getOrDefault("alg_vid_feature_basic_info", new HashMap<>());
|
|
|
+ Future<Integer> future = ThreadPoolFactory.defaultPool().submit(() -> {
|
|
|
+ item.norFeatureMap = getNorVideoFeature(currentMs, vid, userProfile, headInfo, rankInfo, c7Map, c8Map, userOriginInfo, historyVideoMap, videoOriginInfo);
|
|
|
+ return 1;
|
|
|
+ });
|
|
|
+ futures.add(future);
|
|
|
+ }
|
|
|
+
|
|
|
+ try {
|
|
|
+ for (Future<Integer> future : futures) {
|
|
|
+ future.get(1000, TimeUnit.MILLISECONDS);
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("get nor feature error", e);
|
|
|
+ }
|
|
|
}
|
|
|
- return newScore;
|
|
|
}
|
|
|
|
|
|
private double norPowerCalibration(double weight, double exp, double score) {
|
|
@@ -573,20 +504,4 @@ public class RankStrategy4RegionMergeModelV564 extends RankStrategy4RegionMergeM
|
|
|
}
|
|
|
return newScore;
|
|
|
}
|
|
|
-
|
|
|
- private double logRov(double x, double base) {
|
|
|
- if (base > 1) {
|
|
|
- return Math.log(1 + x) / Math.log(base);
|
|
|
- } else {
|
|
|
- return x;
|
|
|
- }
|
|
|
- }
|
|
|
-
|
|
|
- private double logVor(double weight, double x, double base) {
|
|
|
- if (base > 1) {
|
|
|
- return weight * Math.log(1 + x / 10) / Math.log(base);
|
|
|
- } else {
|
|
|
- return x;
|
|
|
- }
|
|
|
- }
|
|
|
}
|