We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent d188b8b commit b448f47Copy full SHA for b448f47
mllib/src/main/scala/org/apache/spark/mllib/tree/impl/DecisionTreeMetadata.scala
@@ -107,8 +107,11 @@ private[tree] object DecisionTreeMetadata extends Logging {
107
numTrees: Int,
108
featureSubsetStrategy: String): DecisionTreeMetadata = {
109
110
- val numFeatures = input.take(1)(0).features.size
111
val numExamples = input.count()
+ require(numExamples > 0, s"DecisionTree requires size of input RDD > 0, " +
112
+ s"but was given by an empty one.")
113
+
114
+ val numFeatures = input.take(1)(0).features.size
115
val numClasses = strategy.algo match {
116
case Classification => strategy.numClasses
117
case Regression => 0
0 commit comments