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