Max/min nodes per group statistics don't make sense if

no groups of nodes are procesed; updated log statements to reflect this
apache · Jun 24, 2016 · f5a6893 · f5a6893
1 parent bd7d24d
commit f5a6893
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/impl/RandomForest.scala
@@ -213,9 +213,9 @@ private[spark] object RandomForest extends Logging {
 
     // Print out runtime statistics
     logInfo(s"Processed $numGroups groups of nodes")
-    logInfo(s"Max nodes per group: $maxNodesPerGroup")
-    logInfo(s"Min nodes per group: $minNodesPerGroup")
     if (numGroups > 0) {
+      logInfo(s"Max nodes per group: $maxNodesPerGroup")
+      logInfo(s"Min nodes per group: $minNodesPerGroup")      
       logInfo(s"Average nodes per group: ${totalNodesProcessed / numGroups.toDouble}")
     }