瀏覽代碼

cross feature bug repair

sunmingze 1 年之前
父節點
當前提交
2a9bc77454
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      src/main/java/examples/sparksql/SparkAdCTRSampleLoader.java

+ 2 - 1
src/main/java/examples/sparksql/SparkAdCTRSampleLoader.java

@@ -70,7 +70,7 @@ public class SparkAdCTRSampleLoader {
         // 转化成bytes
         AdRequestContextBytesFeature adRequestContextBytesFeature = new AdRequestContextBytesFeature(requestContext);
         UserAdBytesFeature userBytesFeature = new UserAdBytesFeature(userFeature);
-        AdItemBytesFeature  adItemBytesFeature = new AdItemBytesFeature(itemFeature);
+        AdItemBytesFeature adItemBytesFeature = new AdItemBytesFeature(itemFeature);
 
         // 特征抽取
         VlogAdCtrLRFeatureExtractor bytesFeatureExtractor;
@@ -79,6 +79,7 @@ public class SparkAdCTRSampleLoader {
         bytesFeatureExtractor.getUserFeatures(userBytesFeature);
         bytesFeatureExtractor.getItemFeature(adItemBytesFeature);
         bytesFeatureExtractor.getContextFeatures(adRequestContextBytesFeature);
+        bytesFeatureExtractor.getCrossFeature(adItemBytesFeature, adRequestContextBytesFeature, userBytesFeature);
 
         ListMultimap<FeatureGroup, BaseFeature> featureMap = bytesFeatureExtractor.getFeatures();
         return parseSamplesToString(label, featureMap);