더북(TheBook)

이제 모델을 생성할 텐데, 배치 정규화가 적용된 모델과 비교하고자 배치 정규화가 적용되지 않는 모델을 생성해 보겠습니다.

코드 8-6 배치 정규화가 적용되지 않은 모델 생성

from tensorflow.keras.models import Sequential
from tensorflow.keras.layers import Dense, BatchNormalization

model1 = Sequential([
    Dense(64, input_shape=(4,), activation="relu"), ------입력층은 (4,0)의 형태를 가지며, 유닛 64개로 구성
    Dense(128, activation='relu'), ------ 입력층 다음으로 유닛 128개를 갖는 밀집층을 두 개 구성
    Dense(128, activation='relu'),
    Dense(64, activation='relu'), ------ 유닛 64개로 구성된 밀집층이 두 개 더 있으며, 렐루(ReLU) 활성화 함수 사용
    Dense(64, activation='relu'),
    Dense(3, activation='softmax') ------ 출력층은 유닛 세 개로 구성되며, 소프트맥스를 활성화 함수로 사용
]);
model1.summary()
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.