Skip to content

Commit 70799f5

Browse files
author
Niketan Pansare
committed
[SYSTEMML-1005] Fix nnz computation issue for mllearn converters
1 parent 4e0949b commit 70799f5

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/main/scala/org/apache/sysml/api/ml/BaseSystemMLClassifier.scala

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -101,6 +101,7 @@ trait BaseSystemMLClassifier extends BaseSystemMLEstimator {
101101
val ml = new MLContext(sc)
102102
val revLabelMapping = new java.util.HashMap[Int, String]
103103
PredictionUtils.fillLabelMapping(y_mb, revLabelMapping)
104+
y_mb.recomputeNonZeros();
104105
val ret = getTrainingScript(isSingleNode)
105106
val script = ret._1.in(ret._2, X_mb).in(ret._3, y_mb)
106107
(ml.execute(script), revLabelMapping)

0 commit comments

Comments
 (0)