File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
mllib/src/main/scala/org/apache/spark/mllib/tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments