Print timing info to logDebug.

xiliu82 · Aug 14, 2014 · d036089 · d036089
1 parent e66f1b1
commit d036089
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala
@@ -167,7 +167,8 @@ class DecisionTree (private val strategy: Strategy) extends Serializable with Lo
 
     timer.stop("total")
 
-    //println(timer)  // Print internal timing info.
+    logDebug("Internal timing for DecisionTree:")
+    logDebug(s"$timer")
 
     new DecisionTreeModel(topNode, strategy.algo)
   }

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/tree/impl/TimeTracker.scala b/mllib/src/main/scala/org/apache/spark/mllib/tree/impl/TimeTracker.scala
@@ -67,9 +67,8 @@ class TimeTracker extends Serializable {
    * Print all timing results in seconds.
    */
   override def toString: String = {
-    s"Timing\n" +
-      totals.map { case (label, elapsed) =>
-          s"  $label: ${elapsed / 1e9}"
-        }.mkString("\n")
+    totals.map { case (label, elapsed) =>
+        s"  $label: ${elapsed / 1e9}"
+      }.mkString("\n")
   }
 }