Skip to content

Commit 201702f

Browse files
committed
making some more methods private
1 parent f963ef5 commit 201702f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mllib/src/main/scala/org/apache/spark/mllib/tree/DecisionTree.scala

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1114,7 +1114,7 @@ object DecisionTree extends Serializable with Logging {
11141114
/**
11151115
* Calculates the classifier accuracy.
11161116
*/
1117-
def accuracyScore(model: DecisionTreeModel, data: RDD[LabeledPoint],
1117+
private def accuracyScore(model: DecisionTreeModel, data: RDD[LabeledPoint],
11181118
threshold: Double = 0.5): Double = {
11191119
def predictedValue(features: Array[Double]) = {
11201120
if (model.predict(features) < threshold) 0.0 else 1.0
@@ -1130,7 +1130,7 @@ object DecisionTree extends Serializable with Logging {
11301130
/**
11311131
* Calculates the mean squared error for regression.
11321132
*/
1133-
def meanSquaredError(tree: DecisionTreeModel, data: RDD[LabeledPoint]): Double = {
1133+
private def meanSquaredError(tree: DecisionTreeModel, data: RDD[LabeledPoint]): Double = {
11341134
data.map { y =>
11351135
val err = tree.predict(y.features) - y.label
11361136
err * err

0 commit comments

Comments
 (0)