diff options
Diffstat (limited to 'mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala')
-rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala index 5d11ed0971..a4287483d1 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/tree/Split.scala @@ -75,7 +75,7 @@ private[tree] object Split { * @param numCategories Number of categories for this feature. */ @DeveloperApi -final class CategoricalSplit private[ml] ( +class CategoricalSplit private[ml] ( override val featureIndex: Int, _leftCategories: Array[Double], @Since("2.0.0") val numCategories: Int) @@ -160,7 +160,7 @@ final class CategoricalSplit private[ml] ( * Otherwise, it goes right. */ @DeveloperApi -final class ContinuousSplit private[ml] (override val featureIndex: Int, val threshold: Double) +class ContinuousSplit private[ml] (override val featureIndex: Int, val threshold: Double) extends Split { override private[ml] def shouldGoLeft(features: Vector): Boolean = { |