fix the Float not showing correctly problem (#13617)

Merge this PR for 1.4.x
apache · Dec 11, 2018 · 1f8bb26 · 1f8bb26
1 parent 449e17d
commit 1f8bb26
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/scala-package/infer/src/main/scala/org/apache/mxnet/infer/javaapi/Predictor.scala b/scala-package/infer/src/main/scala/org/apache/mxnet/infer/javaapi/Predictor.scala
@@ -80,10 +80,11 @@ class Predictor private[mxnet] (val predictor: org.apache.mxnet.infer.Predictor)
                               An extra List is needed for when the model has more than one input.
     * @return                  Indexed sequence array of outputs
     */
-  def predict(input: java.util.List[java.util.List[Float]]):
-  java.util.List[java.util.List[Float]] = {
+  def predict(input: java.util.List[java.util.List[java.lang.Float]]):
+  java.util.List[java.util.List[java.lang.Float]] = {
     val in = JavaConverters.asScalaIteratorConverter(input.iterator).asScala.toIndexedSeq
-    (predictor.predict(in map {a => a.asScala.toArray}) map {b => b.toList.asJava}).asJava
+    (predictor.predict(in map {a => a.asScala.map(Float2float).toArray})
+      map {b => b.map(float2Float).toList.asJava}).asJava
   }