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 <[email protected]>
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
}