Addressing reviewers comments: unnecessary copy of data in predict

apache · Jul 24, 2015 · 9d18469 · 9d18469
1 parent 35125ab
commit 9d18469
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/ann/Layer.scala b/mllib/src/main/scala/org/apache/spark/mllib/ann/Layer.scala
@@ -599,7 +599,8 @@ class FeedForwardModel private(val layerModels: Array[LayerModel],
   }
 
   override def predict(data: Vector): Vector = {
-    val result = forward(data.toBreeze.toDenseVector.toDenseMatrix.t)
+    val size = data.size
+    val result = forward(new BDM[Double](size, 1, data.toArray))
     Vectors.dense(result.last.toArray)
   }
 }