K-평균 군집화를 대략적으로 살펴보았으므로 이제 파이토치에서 사용하는 방법을 예제로 확인해 봅시다. 미리 말해 두지만 파이토치는 딥러닝을 위한 프레임워크이기 때문에 머신 러닝 기법들을 간편하게 사용할 수 있는 방법들을 제공하지 않습니다. 파이토치에서는 머신 러닝 알고리즘이 수행해야 할 일들을 개발자가 일일이 함수로 구현해 주어야 합니다. 이러한 과정이 번거롭다면 간편하게 사이킷런(scikit-learn) 라이브러리를 사용해야 합니다. 그럼에도 파이토치에서 머신 러닝 기법에 대한 사용 방법을 알아보는 이유는 여러분이 인터넷이나 논문을 통해 파이토치로 구현된 머신 러닝 알고리즘을 마주했을 때 부담 없이 코드를 받아들일 수 있도록 돕기 위해서입니다.

파이토치로 K-평균 군집화 예제 구현하기

파이토치에서 K-평균 군집화 알고리즘을 사용하여 데이터를 분류하는 방법을 살펴보겠습니다.

먼저 K-평균 군집화를 사용하기 위한 라이브러리를 설치해야 합니다. 아나콘다 프롬프트에서 다음 명령을 실행해 주세요.

> pip install kmeans_pytorch

kmeans_pytorch 라이브러리는 파이토치에서 K-평균 군집화를 손쉽게 구현할 수 있도록 제공합니다. 하지만 파이토치에서 모든 유형의 머신 러닝을 사용할 수 있는 라이브러리가 있는 것은 아닙니다.