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 {
1114
1114
/**
1115
1115
* Calculates the classifier accuracy.
1116
1116
*/
1117
- def accuracyScore (model : DecisionTreeModel , data : RDD [LabeledPoint ],
1117
+ private def accuracyScore (model : DecisionTreeModel , data : RDD [LabeledPoint ],
1118
1118
threshold : Double = 0.5 ): Double = {
1119
1119
def predictedValue (features : Array [Double ]) = {
1120
1120
if (model.predict(features) < threshold) 0.0 else 1.0
@@ -1130,7 +1130,7 @@ object DecisionTree extends Serializable with Logging {
1130
1130
/**
1131
1131
* Calculates the mean squared error for regression.
1132
1132
*/
1133
- def meanSquaredError (tree : DecisionTreeModel , data : RDD [LabeledPoint ]): Double = {
1133
+ private def meanSquaredError (tree : DecisionTreeModel , data : RDD [LabeledPoint ]): Double = {
1134
1134
data.map { y =>
1135
1135
val err = tree.predict(y.features) - y.label
1136
1136
err * err
You can’t perform that action at this time.
0 commit comments