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로 바뀌었습니다.

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