j

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

@@ -90,7 +90,7 @@ public class I2IDSSMPredict {
 
                         // 分割left_features_str并转换为float数组
                         String[] leftFeaturesArray = leftFeaturesStr.split(",");
-                        double[] leftFeatures = new double[leftFeaturesArray.length];
+                        float[] leftFeatures = new float[leftFeaturesArray.length];
                         for (int i = 0; i < leftFeaturesArray.length; i++) {
                             leftFeatures[i] = Float.parseFloat(leftFeaturesArray[i]);
                         }
@@ -99,7 +99,6 @@ public class I2IDSSMPredict {
                         // 2 设置输入
                         inHandle.reshape(2, new int[]{1, 157});
                         inHandle.copyFromCpu(leftFeatures);
-
                         // 3 预测
                         predictor.run();