Make predict private

feynmanliang · Jul 1, 2015 · c2df35d · c2df35d
1 parent 93aa2ff
commit c2df35d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mllib/src/main/scala/org/apache/spark/ml/clustering/KMeans.scala b/mllib/src/main/scala/org/apache/spark/ml/clustering/KMeans.scala
@@ -122,6 +122,7 @@ class KMeansModel private[ml] (
     validateAndTransformSchema(schema)
   }
 
+  private[clustering]
   def predict(features: Vector): Int = parentModel.predict(features)
 
   def clusterCenters: Array[Vector] = parentModel.clusterCenters