diff options
Diffstat (limited to 'mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala')
| -rw-r--r-- | mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala index b5cb378829..f71d28cf59 100644 --- a/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala +++ b/mllib/src/main/scala/org/apache/spark/ml/tree/Node.scala @@ -115,7 +115,7 @@ private[ml] object Node { * @param impurity Impurity measure at this node (for training data) */ @DeveloperApi -final class LeafNode private[ml] ( +class LeafNode private[ml] ( override val prediction: Double, override val impurity: Double, override private[ml] val impurityStats: ImpurityCalculator) extends Node { @@ -158,7 +158,7 @@ final class LeafNode private[ml] ( * @param split Information about the test used to split to the left or right child. */ @DeveloperApi -final class InternalNode private[ml] ( +class InternalNode private[ml] ( override val prediction: Double, override val impurity: Double, val gain: Double, |