init pr

WeichenXu123 · WeichenXu123 · commit ea062253086c · 2017-09-02T12:30:01.000+08:00
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
@@ -304,7 +304,7 @@ class KMeans @Since("1.5.0") (
   override def fit(dataset: Dataset[_]): KMeansModel = {
     transformSchema(dataset.schema, logging = true)
 
-    val handlePersistence = dataset.rdd.getStorageLevel == StorageLevel.NONE
+    val handlePersistence = dataset.storageLevel == StorageLevel.NONE
     val instances: RDD[OldVector] = dataset.select(col($(featuresCol))).rdd.map {
       case Row(point: Vector) => OldVectors.fromML(point)
     }

Original file line number	Diff line number	Diff line change
`@@ -304,7 +304,7 @@ class KMeans @Since("1.5.0") (`
`304`	`304`	`override def fit(dataset: Dataset[_]): KMeansModel = {`
`305`	`305`	`transformSchema(dataset.schema, logging = true)`
`306`	`306`
`307`		`- val handlePersistence = dataset.rdd.getStorageLevel == StorageLevel.NONE`
	`307`	`+ val handlePersistence = dataset.storageLevel == StorageLevel.NONE`
`308`	`308`	`val instances: RDD[OldVector] = dataset.select(col($(featuresCol))).rdd.map {`
`309`	`309`	`case Row(point: Vector) => OldVectors.fromML(point)`
`310`	`310`	`}`