Repository: incubator-systemml Updated Branches: refs/heads/master 4e0949bae -> 70799f521
[SYSTEMML-1005] Fix nnz computation issue for mllearn converters Project: http://git-wip-us.apache.org/repos/asf/incubator-systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-systemml/commit/70799f52 Tree: http://git-wip-us.apache.org/repos/asf/incubator-systemml/tree/70799f52 Diff: http://git-wip-us.apache.org/repos/asf/incubator-systemml/diff/70799f52 Branch: refs/heads/master Commit: 70799f521fcdfbe8f87ba06ffb48b701ed97c14a Parents: 4e0949b Author: Niketan Pansare <npan...@us.ibm.com> Authored: Tue Oct 4 12:56:19 2016 -0700 Committer: Niketan Pansare <npan...@us.ibm.com> Committed: Tue Oct 4 12:56:19 2016 -0700 ---------------------------------------------------------------------- src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala | 1 + 1 file changed, 1 insertion(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-systemml/blob/70799f52/src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala ---------------------------------------------------------------------- diff --git a/src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala b/src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala index 87046b3..539c2c1 100644 --- a/src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala +++ b/src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala @@ -101,6 +101,7 @@ trait BaseSystemMLClassifier extends BaseSystemMLEstimator { val ml = new MLContext(sc) val revLabelMapping = new java.util.HashMap[Int, String] PredictionUtils.fillLabelMapping(y_mb, revLabelMapping) + y_mb.recomputeNonZeros(); val ret = getTrainingScript(isSingleNode) val script = ret._1.in(ret._2, X_mb).in(ret._3, y_mb) (ml.execute(script), revLabelMapping)