스파크에서 클러스터 센터 초기화

Question

나는이 질문과 매우 비슷한 것을하려고합니다 : How to initialize cluster centers for K-means in Spark MLlib? 그러나 나는 완전히 해결책을 이해하지 못합니다. 좀 더 무게 중심을 추가하려고하면, 나는 오류 얻을 :

Exception in thread "main" java.lang.IllegalArgumentException: requirement failed: mismatched cluster count

나는 기본적으로

val initialModel = new KMeansModel(
    Array("[0.6, 0.6, 5.0]", "[8.0, 8.0, 1.0]", "[11, 9.0, 7.0]").map(Vectors.parse(_)) 
) 
val model = new KMeans() 
    .setInitialModel(initialModel) 
    .setK(3) 
    .run(data) 

Answer 1

사용을 KMeans 그것은 초기를 설정하기 전에 K를 설정하는 것만 큼 쉽다 2로 K 설정 모델 (KMeansModel.k 및 KMeans.k는 일치해야 함).

val initialModel = new KMeansModel(
    Array("[0.6, 0.6, 5.0]", "[8.0, 8.0, 1.0]", "[11, 9.0, 7.0]").map(Vectors.parse(_)) 
) 

val model = new KMeans() 
    .setK(3) 
    .setInitialModel(initialModel) 
    .run(data) 

의