TitleConvert.java 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263
  1. package examples.myUtils;
  2. import com.tzld.piaoquan.recommend.similarity.word2vec.Segment;
  3. import java.util.HashMap;
  4. import java.util.List;
  5. import java.util.Map;
  6. public class TitleConvert {
  7. public static String getFeature(Map<String, String> record, List<Map<String, String>> videoSeq, int scale) {
  8. String result = "";
  9. try {
  10. Map<String, String> headInfo = ConvertUtils.getRecordCol(record, "v2_feature");
  11. String title = headInfo.getOrDefault("title", "");
  12. if (!title.isEmpty()) {
  13. List<String> words = Segment.getWords(title);
  14. result = String.join(" ", words);
  15. }
  16. } catch (Exception e) {
  17. e.printStackTrace();
  18. }
  19. return result;
  20. }
  21. private static Map<String, Map<String, String>> getUserOriginInfo(Map<String, String> record) {
  22. Map<String, Map<String, String>> map = new HashMap<>();
  23. map.put("mid_global_feature_20250212", ConvertUtils.getRecordCol(record, "c1_feature"));
  24. map.put("alg_mid_feature_return_tags", ConvertUtils.getRecordCol(record, "c5_feature"));
  25. map.put("alg_mid_feature_share_tags", ConvertUtils.getRecordCol(record, "c6_feature"));
  26. map.put("alg_mid_feature_sharecf", new HashMap<>());
  27. map.put("alg_mid_feature_returncf", new HashMap<>());
  28. map.put("alg_recsys_feature_user_share_return_stat", ConvertUtils.getRecordCol(record, "c9_feature"));
  29. return map;
  30. }
  31. private static Map<String, Map<String, Map<String, String>>> getVideoOriginInfo(Map<String, String> record) {
  32. Map<String, Map<String, String>> map = new HashMap<>();
  33. map.put("alg_recsys_feature_video_clean_stat", ConvertUtils.getRecordCol(record, "b0_feature"));
  34. map.put("alg_vid_global_feature_20250212", ConvertUtils.getRecordCol(record, "b1_feature"));
  35. map.put("alg_vid_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b2_feature"));
  36. map.put("alg_vid_recommend_flowpool_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b3_feature"));
  37. map.put("alg_vid_apptype_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b4_feature"));
  38. map.put("alg_vid_province_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b5_feature"));
  39. map.put("alg_vid_brand_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b6_feature"));
  40. map.put("alg_vid_hotsencetype_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b7_feature"));
  41. map.put("alg_merge_cate1_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b8_feature"));
  42. map.put("alg_merge_cate2_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b9_feature"));
  43. map.put("alg_channel_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b10_feature"));
  44. map.put("alg_festive_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b11_feature"));
  45. map.put("alg_video_unionid_recommend_exp_feature_20250212", ConvertUtils.getRecordCol(record, "b13_feature"));
  46. map.put("scene_type_vid_cf_feature_20250212", ConvertUtils.getRecordCol(record, "d1_feature"));
  47. map.put("vid_click_cf_feature_20250212", ConvertUtils.getRecordCol(record, "d2_feature"));
  48. map.put("alg_recsys_feature_cf_i2i_v2", ConvertUtils.getRecordCol(record, "d3_feature"));
  49. map.put("alg_recsys_feature_video_recommend_channel_layer", ConvertUtils.getRecordCol(record, "b14_feature"));
  50. map.put("alg_recsys_feature_video_recommend_channel_layer_head", ConvertUtils.getRecordCol(record, "b15_feature"));
  51. Map<String, Map<String, Map<String, String>>> allMap = new HashMap<>();
  52. String vid = record.get("vid");
  53. allMap.put(vid, map);
  54. return allMap;
  55. }
  56. }