Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
minmingzhu committed Jul 16, 2024
1 parent 2745466 commit 162e8cb
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ import org.apache.spark.ml.util._
import org.apache.spark.mllib.clustering.{KMeansModel => MLlibKMeansModel}
import org.apache.spark.mllib.linalg.{Vector => OldVector, Vectors => OldVectors}
import org.apache.spark.rdd.RDD
import java.time.Instant

class KMeansDALImpl(var nClusters: Int,
var maxIterations: Int,
Expand All @@ -41,7 +42,7 @@ class KMeansDALImpl(var nClusters: Int,
val metrics_name = "Kmeans_" + executorNum
val kmeansTimer = new Utils.AlgoTimeMetrics(metrics_name, sparkContext)
val useDevice = sparkContext.getConf.get("spark.oap.mllib.device", Utils.DefaultComputeDevice)
val storePath = sparkContext.getConf.get("spark.oap.mllib.kvsStorePath")
val storePath = sparkContext.getConf.get("spark.oap.mllib.kvsStorePath") + "/" + Instant.now()
val computeDevice = Common.ComputeDevice.getDeviceByName(useDevice)
kmeansTimer.record("Preprocessing")

Expand Down

0 comments on commit 162e8cb

Please sign in to comment.