浏览代码

添加nor模型特征和分桶

jch 4 月之前
父节点
当前提交
4f92844125

+ 5 - 0
recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/score/NorXGBRegressionScorer.java

@@ -145,6 +145,11 @@ public class NorXGBRegressionScorer extends XGBRegressionModelScorer {
                 LOGGER.error("score error for doc={} exception={}", item.getVideoId(), ExceptionUtils.getFullStackTrace(e));
             }
         }
+        if (pro < 1E-8) {
+            pro = 0;
+        } else if (pro > 50) {
+            pro = 50;
+        }
         item.getScoresMap().put("NorXGBScore", pro);
         return pro;
     }

+ 1 - 1
recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/score/ScorerPipeline.java

@@ -213,7 +213,7 @@ public class ScorerPipeline {
             final String scorerName = scorerNames.length > 0 ? scorerNames[scorerNames.length - 1] : fullScorerName;
 
             final List<RankItem> scoreRankerItems = items;
-            Map<String, String> tmpUserFeatMap = userFeatMap;
+            final Map<String, String> tmpUserFeatMap = userFeatMap;
             Callable<List<RankItem>> callable = () -> scorer.scoring(sceneFeatureMap, tmpUserFeatMap, scoreRankerItems);
 
             // execute score use thread to protected score worst time