소스 검색

save model

丁云鹏 4 달 전
부모
커밋
2e55992a0c
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 2 2
      recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DSSMModel.java

+ 2 - 2
recommend-model-service/src/main/java/com/tzld/piaoquan/recommend/model/service/model/DSSMModel.java

@@ -34,8 +34,8 @@ public class DSSMModel implements Model {
             log.info("predictor2 outName{}={}", i, predictor.getOutputNameById(i));
         }
         // 2 设置输入
-        inHandle.reshape(2, new int[]{1, 47});
-        float[] inData = new float[1 * 47];
+        inHandle.reshape(2, new int[]{1, 157});
+        float[] inData = new float[1 * 157];
         inHandle.copyFromCpu(inData);
 
         // 3 预测