瀏覽代碼

save model

丁云鹏 4 月之前
父節點
當前提交
2e55992a0c

+ 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 预测