Bladeren bron

feat:xgboost

zhaohaipeng 8 maanden geleden
bovenliggende
commit
af8fe19084

+ 1 - 1
ad-engine-commons/src/main/java/com/tzld/piaoquan/ad/engine/commons/score/model/XGBoostModel.java

@@ -565,7 +565,7 @@ public class XGBoostModel extends Model {
                 values[i] = v;
             }
             DMatrix dm = new DMatrix(values, 1, features.length, 0.0f);
-            float[][] result = model._booster().predict(dm, false, 100);
+            float[][] result = model._booster().predict(dm, false, 0);
             return result[0][0];
         } catch (Exception e) {
             return 0f;

+ 1 - 1
ad-engine-service/src/main/java/com/tzld/piaoquan/ad/engine/service/score/impl/RankServiceImpl.java

@@ -93,7 +93,7 @@ public class RankServiceImpl implements RankService {
         List<AdRankItem> adRankItems = fmRankService.adItemRank(request, scoreParam);
         for (AdRankItem adRankItem : adRankItems) {
             JSONObject json = new JSONObject();
-            json.put("scoreMap", adRankItem);
+            json.put("scoreMap", adRankItem.getScoreMap());
             json.put("allfeature", adRankItem.getFeatureMap());
             json.put("metafeaturemap", adRankItem.getMetaFeatureMap());
             json.put("cid", adRankItem.getId());