dssm train

recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IDSSMPredict.java

@@ -105,7 +105,7 @@ public class I2IDSSMPredict {
                         // 4 获取输入Tensor
                         String outNames = predictor.getOutputNameById(0);
                         Tensor outHandle = predictor.getOutputHandle(outNames);
-                        double[] outData = new double[outHandle.getSize()];
+                        float[] outData = new float[outHandle.getSize()];
                         outHandle.copyToCpu(outData);