더북(TheBook)

이제 그림 3-16의 신경망 아키텍처를 따라 Sequential 모델을 만든다.

from keras.models import Sequential
from keras.layers import Dense
 
model = Sequential()
model.add(Dense(128, activation='relu', input_dim=X_train.shape[1]))
model.add(Dense(64, activation='relu'))
model.add(Dense(32, activation='relu'))
model.add(Dense(8, activation='relu'))
model.add(Dense(1))

모델 훈련을 시작하기 전에 모델 구조를 검증하는 게 좋다.

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