더북(TheBook)

생성한 모델에 대해 컴파일을 적용하고, fit() 메서드를 사용하여 모델을 학습시켜 보겠습니다.

코드 8-7 모델 훈련

model1.compile(
    optimizer='adam',
    loss='categorical_crossentropy',
    metrics=['accuracy']
) ------ ①

history1 = model1.fit(
    X_train,
    y_train,
    epochs=1000,
    validation_split=0.25,
    batch_size=40,
    verbose=2
)

① 옵티마이저로 아담(Adam) 알고리즘을 적용하고, 손실 함수로 크로스엔트로피(categorical_ crossentropy)를 적용합니다.

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