Check the actual model

apache · Apr 2, 2018 · e0f9b09 · e0f9b09
1 parent 3d529f8
commit e0f9b09
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala b/mllib/src/test/scala/org/apache/spark/ml/clustering/KMeansSuite.scala
@@ -226,7 +226,7 @@ class KMeansSuite extends SparkFunSuite with MLlibTestSparkContext with DefaultR
       val pmmlClusteringModel = pmml.getModels.get(0).asInstanceOf[ClusteringModel]
       assert(pmmlClusteringModel.getNumberOfClusters === clusterCenters.length)
     }
-    testPMMLWrite(sc, model, checkModel)
+    testPMMLWrite(sc, kmeansModel, checkModel)
   }
 }