zhaohaipeng 1 mese fa
parent
commit
554fd665e5

+ 7 - 5
recommend-server-service/src/main/java/com/tzld/piaoquan/recommend/server/service/score/model/PAIModel.java

@@ -1,5 +1,6 @@
 package com.tzld.piaoquan.recommend.server.service.score.model;
 
+import com.alibaba.fastjson.JSON;
 import com.aliyun.openservices.eas.predict.http.HttpConfig;
 import com.aliyun.openservices.eas.predict.http.PredictClient;
 import com.aliyun.openservices.eas.predict.proto.PredictProtos;
@@ -77,6 +78,10 @@ public class PAIModel {
         List<String> scoreKeys = Arrays.asList("y_return_n_uv", "probs_is_share");
         int size = rankItems.size();
         String currentName = "";
+        // 生成特征名与特征值数组的映射
+        Map<String, double[]> doubleFeatureArr = new HashMap<>();
+        Map<String, String[]> stringFeatureArr = new HashMap<>();
+
         try {
 
             List<Map<String, Float>> featureMapList = new ArrayList<>(rankItems.size());
@@ -87,9 +92,6 @@ public class PAIModel {
                 featureMapStringList.add(Optional.ofNullable(rankItem.featureMapString).orElse(new HashMap<>()));
             }
 
-            // 生成特征名与特征值数组的映射
-            Map<String, double[]> doubleFeatureArr = new HashMap<>();
-            Map<String, String[]> stringFeatureArr = new HashMap<>();
             for (Tuple4<String, String, String, TFDataType> featureKeyTuple : featureKeys) {
                 TFDataType dataType = featureKeyTuple.getV4();
                 String featureName = featureKeyTuple.getV2();
@@ -127,14 +129,14 @@ public class PAIModel {
             }
             TFResponse response = predictClient.predict(request);
             Map<String, PredictProtos.ArrayProto> outputsMap = response.getOutputsMap();
-            LOGGER.info("response keys: {}", outputsMap.keySet());
+            // LOGGER.info("response keys: {}", outputsMap.keySet());
             Map<String, List<Float>> resultMap = new HashMap<>();
             for (String scoreKey : scoreKeys) {
                 resultMap.put(scoreKey, response.getFloatVals(scoreKey));
             }
             return resultMap;
         } catch (Exception e) {
-            LOGGER.error("pai eas error currentName: {}", currentName, e);
+            LOGGER.error("pai eas error currentName: {}, {}", currentName, JSON.toJSONString(stringFeatureArr), e);
             Map<String, List<Float>> resultMap = new HashMap<>();
             for (String scoreKey : scoreKeys) {
                 resultMap.put(scoreKey, Collections.nCopies(size, 0.0f));