이제 이러한 과정을 지나 다시 앞에서 Dense() 함수를 이용해 만들었던 기본 층에 연결해 볼까요? 이때 주의할 점은 컨볼루션 층이나 맥스 풀링은 주어진 이미지를 2차원 배열인 채로 다룬다는 것입니다. 이를 1차원 배열로 바꾸어 주어야 활성화 함수가 있는 층에서 사용할 수 있습니다. 따라서 Flatten() 함수를 사용해 2차원 배열을 1차원으로 바꾸어 줍니다.

model.add(Flatten())

이를 포함해서 새롭게 구현할 딥러닝 프레임을 그림 16-9와 같이 설정해 보겠습니다.

그림 16-9 | 드롭아웃과 플래튼 추가하기