Ver Fonte

区分特征

xueyiming há 3 dias atrás
pai
commit
d597748926

+ 10 - 6
ad-engine-commons/src/main/java/com/tzld/piaoquan/ad/engine/commons/score/model/PAIModelV1.java

@@ -38,14 +38,18 @@ public class PAIModelV1 {
         client.setModelName("ad_rank_dnn_v11_easyrec");
     }
 
-    private static final String[] sparseStrFeatures = {
+    private static final String[] sparseUserStrFeatures = {
             "brand", "region", "city", "cate1", "cate2", "user_cid_click_list", "user_cid_conver_list",
             "user_vid_return_tags_2h", "user_vid_return_tags_1d", "user_vid_return_tags_3d", "user_vid_return_tags_7d",
             "user_vid_return_tags_14d", "root_source_scene", "root_source_channel", "title_split"
     };
 
-    private static final String[] sparseLongFeatures = {
-            "vid", "apptype", "hour", "hour_quarter", "is_first_layer"
+    private static final String[] sparseUserLongFeatures = {
+            "is_first_layer"
+    };
+
+    private static final String[] sparseSceneLongFeatures = {
+            "vid", "apptype", "hour", "hour_quarter"
     };
 
     private static final String[] sparseAdLongFeatures = {
@@ -117,19 +121,19 @@ public class PAIModelV1 {
         try {
             TFRequest request = new TFRequest();
 
-            for (String feature : sparseStrFeatures) {
+            for (String feature : sparseUserStrFeatures) {
                 String key = feature.replace("_x_", "*").replace("_view", "(view)");
                 String v = userFeatureMap.getOrDefault(key, "");
                 request.addFeed(feature, TFDataType.DT_STRING, new long[]{1}, new String[]{v});
             }
 
-            for (String feature : sparseLongFeatures) {
+            for (String feature : sparseUserLongFeatures) {
                 String key = feature.replace("_x_", "*").replace("_view", "(view)");
                 long v = NumberUtils.toLong(userFeatureMap.getOrDefault(key, "0"), 0);
                 request.addFeed(feature, TFDataType.DT_INT64, new long[]{1}, new long[]{v});
             }
 
-            for (String feature : sparseLongFeatures) {
+            for (String feature : sparseSceneLongFeatures) {
                 String key = feature.replace("_x_", "*").replace("_view", "(view)");
                 long v = NumberUtils.toLong(sceneFeatureMap.getOrDefault(key, "0"), 0);
                 request.addFeed(feature, TFDataType.DT_INT64, new long[]{1}, new long[]{v});