더북(TheBook)

2.7.1.5 시그모이드 활성화 함수

마지막 출력 레이어에는 클래스 레이블을 예측할 수 있는 활성화 함수가 필요하다. 이번 프로젝트에서는 단순히 이진 클래스를 예측한다. 즉, 당뇨 발병 환자라면 1을 출력하고 미발병 환자라면 0을 출력한다. 이진 분류 문제에는 시그모이드 활성화 함수가 적합하다.

시그모이드 활성화 함수의 공식은 다음과 같다.

공식이 복잡해 보여도 의외로 간단하다. 시그모이드 함수는 입력값을 다음 그림 2-15와 같이 0과 1 사이 값으로 압축한다.

▲ 그림 2-15 시그모이드 활성화 함수

예측 결과를 만드는 방법도 간단하다. 시그모이드 함수의 결과가 0.5보다 크면 예측 결과를 1로 출력한다. 반대로 0.5보다 작으면 0을 출력한다. 다시 말해, 시그모이드 활성화 함수는 입력값을 받아 이진 클래스(즉, 1 또는 0)를 출력한다. 따라서 환자의 당뇨 발병 여부를 예측하는 데 사용할 수 있다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.