[incubator-mxnet] branch master updated: fix the Float not showing correctly problem (#13617)

[la...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

lanking pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 1f8bb26  fix the Float not showing correctly problem (#13617)
1f8bb26 is described below

commit 1f8bb26b63f030ce9d64d3a45aae0cc216572de0
Author: Lanking <la...@live.com>
AuthorDate: Tue Dec 11 15:11:27 2018 -0800

    fix the Float not showing correctly problem (#13617)
    
    Merge this PR for 1.4.x
---
 .../src/main/scala/org/apache/mxnet/infer/javaapi/Predictor.scala  | 7 ++++---
 1 file changed, 4 insertions(+), 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
index 8c48742..0466693 100644
--- 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
   }