ML & DL

Classification

안모 2023. 6. 6. 23:03

Classification은 이진분류를 하거나, 분류 대상이 셋 이상일 경우 각 분류에 맞게 대상을 나눈다.

이진 분류는 대상을 둘 중 하나로 분류한다.

  • 0 또는 1
  • True 또는 False 등

대표적으로 메일이 스팸메일인지 스팸메일이 아닌지 분류하는 방식이 이진 분류이다.

 

다중 분류는 대상을 여러 가지 분류 중 하나로 분류한다.

  • 물건 분류
  • 동물 분류 등

사진에 나와 있는 물건이 자동차인지 비행기인지 자전거인지 분류하는 방식이다.

 

다중 분류를 수행하는데 사용되는 함수가 Softmax 함수이다. 0과 1사이의 값을 지니며 도출된 값을 전부 더하면 1이 되는 확률의 조건을 갖춘 함수이다.

  • 자동차: 0.5
  • 비행기: 0.3
  • 자전거: 0.2

위 처럼 값이 나왔을 경우 자동차일 확률이 가장 높으며 확률을 다 더하면 1이 된다.