[MINOR] Additional debug info on matrix compression (column classify)

ravichinoy · Feb 16, 2018 · 72830f0 · 72830f0
1 parent 6a4f1e7
commit 72830f0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/main/java/org/apache/sysml/runtime/compress/CompressedMatrixBlock.java b/src/main/java/org/apache/sysml/runtime/compress/CompressedMatrixBlock.java
@@ -269,7 +269,7 @@ public MatrixBlock compress(int k)
 		CompressedSizeInfo[] sizeInfos = (k > 1) ?
 				computeCompressedSizeInfos(bitmapSizeEstimator, numCols, k) : 
 				computeCompressedSizeInfos(bitmapSizeEstimator, numCols);
-		long nnzUC = 0;		
+		long nnzUC = 0;
 		for (int col = 0; col < numCols; col++)  {
 			double uncompSize = getUncompressedSize(numRows, 1, 
 				OptimizerUtils.getSparsity(numRows, 1, sizeInfos[col].getEstNnz()));
@@ -300,6 +300,15 @@ public MatrixBlock compress(int k)
 			}
 		}
 
+		if( LOG.isTraceEnabled() ) {
+			LOG.trace("C: "+Arrays.toString(colsC.toArray(new Integer[0])));
+			LOG.trace("-- compression ratios: "+Arrays.toString(
+				colsC.stream().map(c -> compRatios.get(c)).toArray()));
+			LOG.trace("UC: "+Arrays.toString(colsUC.toArray(new Integer[0])));
+			LOG.trace("-- compression ratios: "+Arrays.toString(
+				colsUC.stream().map(c -> compRatios.get(c)).toArray()));
+		}
+
 		if( LOG.isDebugEnabled() ) {
 			_stats.timePhase1 = time.stop();
 			LOG.debug("Compression statistics:");