모두의 딥러닝 개정 3판: 4 소프트맥스

4 소프트맥스

이제 모델을 만들어 줄 차례입니다. 다음 코드를 보면서 이전에 실행했던 피마 인디언의 당뇨병 예측과 무엇이 달라졌는지 찾아보기 바랍니다.

# 모델 설정
model = Sequential()
model.add(Dense(12, input_dim=4, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(3, activation='softmax'))
model.summary()

# 모델 컴파일
model.compile(loss='categorical_crossentropy', optimizer='adam',
metrics=['accuracy'])

찾으셨나요? 세 가지가 달라졌습니다. 첫째 출력층의 노드 수가 3으로 바뀌었군요. 그리고 활성화 함수가 softmax로 바뀌었습니다. 마지막으로 컴파일 부분에서 손실 함수 부분이 categorical_crossentropy로 바뀌었습니다.

신간 소식 구독하기

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