|
@@ -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});
|