[기초 1] tensorflow 2 keras 모델 정의

keras 모델 정의는 2가지로 나뉘어져 있다.

 

첫 번째는 Sequential model

두 번째는 tensorflow.keras.Model 을 상속받은 custom model

 

그럼 Dense Layer 로만 구성된 모델로 예제를 구성하겠다.

 

1. Sequential model

import tensorflow as tf
from tensorflow.keras import layers

# make sequential model instance
def makeModel():
	sequential_model = tf.keras.Sequential()
    
    # hidden layer
    sequential_model.add(layers.Dense(64, activation='relu'))
    sequential_model.add(layers.Dense(64, activation='relu'))
    
    # output layer
    sequential_model.add(layers.Dense(10, activation='softmax'))
    
    return sequential_model
    

if __name__ == "__main__":
	my_model = makeModel()

 

2. custom model

import tensorflow as tf
from tensorflow.keras import layers

# custom model
class myModel(tf.keras.Model):
	# make layers in init function
	def __init__(self):
    	super(myModel, self).__init__()
        
        self.hidden_layer_1 = layers.Dense(64, activation='relu')
        self.hidden_layer_2 = layers.Dense(64, activation='relu')
        
        self.output_layer = layers.Dense(10, activation='softmax')
    
    # make graph (flow data)
	def call(self, input):
    	hidden_data_1 = self.hidden_layer_1(input)
        hidden_data_2 = self.hidden_layer_2(input)
        
        output_data = self.output_layer(hidden_data)
        
        return output_data
   
   
if __name__ == "__main__":
	my_model = myModel()

 

둘의 차이는 한눈에도 들어오지만 굳이 꼽자면 크게 두가지이다.

 

첫 번째는

Sequential은 layer를 순서대로 넣어주면 자동적으로 graph가 그려지는 것에 반해,

custom model은 직접 call function 안에서 graph를 그려줘야 한다.

 

여기서 graph를 그린다는 의미는 데이터의 흐름도라고 생각하면 편하다.

 

아래 링크에서 그래프를 시각화한 것을 잘 표현하였으므로 참조하면 이해하기 편하다.

 

https://tensorflowkorea.gitbooks.io/tensorflow-kr/content/g3doc/how_tos/graph_viz/

그래프 시각화 · 텐서플로우 문서 한글 번역본

두 번째는

Sequential은 특별한 function, class 정의가 필요없이 바로 instance 생성이 가능하지만, 

custom model은 class로 상속받아 새로 정의한 후 instance를 생성해야 한다.

 

자세한 용어 설명 참조

딥러닝 용어

Sequential

https://www.tensorflow.org/guide/keras/overview?hl=ko#sequential_모델

케라스: 빠르게 훑어보기  |  TensorFlow Core

 

tensorflow.keras.Model

https://www.tensorflow.org/api_docs/python/tf/keras/Model

tf.keras.Model  |  TensorFlow Core v2.1.0

 

 

컴퓨터 용어

 

상속

https://wikidocs.net/16073

위키독스

 

super 사용법

https://bluese05.tistory.com/5

python super() 함수와 2.X 와 3.X 사용법