소스 검색

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