반응형
본 포스팅은 다음 과정을 정리 한 글입니다.
Custom Models, Layers, and Loss Functions with TensorFlow
www.coursera.org/specializations/tensorflow-advanced-techniques
다음은 지난 시간에 배웠던 Functional API를 함수화하여 만든 코드입니다.
다른 점은 tf.keras가 앞에 붙었다는 점?! layer들에는 tf.keras.layers가 붙었고 각각 변수들에 이름이 설정되어 있습니다.
아래 코드는 first_dense layer를 128개의 뉴런들을 가진 Dense layer로 정의하는 코드로 flatten_layer다음에 오게 하겠다는 의미입니다.
위의 코드를 아래와 같이도 변경할 수 있습니다. 아래는 오타같은데 first_Dense가아니고 first_dense입니다. 저렇게 두줄로 나눠서 호출도 가능합니다.
여러분은 두가지 방법 중에 어떤 방법이 편하신가요?
저는 첫 번째가 더 코드가 깔끔한 것 같네요 ㅎㅎ 첫 번째 방식으로 코딩을 하구요!
다음 시간에는 더 복잡한 model 구조를 만들어 보도록 하겠습니다~
감사합니다.
300x250
댓글