xueyiming 3 mēneši atpakaļ
vecāks
revīzija
af3218e0a3

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

@@ -63,7 +63,7 @@ public class PAIModel {
             for (int i = 0; i < items.size(); i++) {
                 for (String feature : itemFeatures) {
                     String key = feature.replace("_x_", "*").replace("_view", "(view)");
-                    double[] doubles = feed.computeIfAbsent(feature, k -> new double[itemFeatures.length]);
+                    double[] doubles = feed.computeIfAbsent(feature, k -> new double[items.size()]);
                     if (MapUtils.isEmpty(items.get(i).getFeatureMap())) {
                         doubles[i] = 0.0;
                         continue;
@@ -73,6 +73,7 @@ public class PAIModel {
                 }
             }
             for (Map.Entry<String, double[]> entry : feed.entrySet()) {
+                LOGGER.info("entry key={}, value.size={}", entry.getKey(), entry.getValue().length);
                 request.addFeed(entry.getKey(), TFDataType.DT_DOUBLE, new long[]{items.size()}, entry.getValue());
             }
             request.addFetch("probs");