Rebase mater branch and add copy methods

rxin · Jul 1, 2015 · f80319a · f80319a
1 parent 85d92b1
commit f80319a
Showing 1 changed file with 8 additions 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
@@ -107,6 +107,12 @@ class KMeansModel private[ml] (
   val paramMap: ParamMap,
   val parentModel: mllib.clustering.KMeansModel
 ) extends Model[KMeansModel] with KMeansParams {
+
+  override def copy(extra: ParamMap): KMeansModel = {
+    val copied = new KMeansModel(uid, paramMap, parentModel)
+    copyValues(copied, extra)
+  }
+
   /**
    * Transforms the input dataset.
    */
@@ -145,6 +151,8 @@ class KMeans(override val uid: String) extends Estimator[KMeansModel] with KMean
   setEpsilon(1e-4)
   setSeed(Utils.random.nextLong())
 
+  override def copy(extra: ParamMap): Estimator[KMeansModel] = defaultCopy(extra)
+
   def this() = this(Identifiable.randomUID("kmeans"))
 
   /** @group setParam */