참고로 클러스터 개수(K 개수)를 결정하는 것은 쉽지 않습니다. 클러스터 개수를 좀 더 편리하게 결정할 수 있는 방법은 다음과 같이 클러스터 개수와 WCSS(Within Cluster Sum of Squares) 간 관계를 그래프로 표현한 후, WCSS 변경이 평평하게 하락하는 구간을 선택하는 것입니다(다음 그래프에서는 3이 됩니다).
▲ 그림 11-5 클러스터 개수와 WCSS
K-평균 군집화를 대략적으로 살펴보았으므로 이제 딥러닝을 적용하는 방법을 예제로 확인해 봅시다. 예제는 사전 훈련된 MobileNet 모델1의 가중치를 사용하여 훈련합니다.
1 MobileNet은 컴퓨터 성능이 제한되거나 배터리 성능이 중요한 곳에서 사용될 목적으로 설계된 CNN 구조입니다. 즉, AlexNet처럼 비대한 크기의 네트워크보다는 빠른 성능이 필요한 곳에서 MobileNet을 사용합니다.