dssm train

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

@@ -9,6 +9,7 @@ import com.tzld.piaoquan.recommend.model.produce.service.OSSService;
 import com.tzld.piaoquan.recommend.model.produce.util.CompressUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang.math.NumberUtils;
+import org.apache.commons.lang3.StringUtils;
 import org.apache.hadoop.io.compress.GzipCodec;
 import org.apache.spark.api.java.JavaRDD;
 import org.apache.spark.api.java.JavaSparkContext;
@@ -111,7 +112,8 @@ public class I2IDSSMPredict {
                         float[] outData = new float[outHandle.getSize()];
                         outHandle.copyToCpu(outData);
 
-                        String result = vid + "\t" + outData[0];
+
+                        String result = vid + "\t[" + StringUtils.join(outData, ',') + "]";
 
                         outHandle.destroyNativeTensor();
                         inHandle.destroyNativeTensor();