Model Subclassing API를 이용하는 방법
본질적으로는 Functional API와 차이가 없지만, Model Subclassing API는 다음 예시 코드처럼 사용자가 가장 자유롭게 모델을 구축할 수 있는 방법입니다.
class ResNet(tf.keras.Model):
def __init__(self):
super(ResNet, self).__init__()
self.block_1 = layers.Dense(32, activation='relu')
self.block_2 = layers.Dense(num_classes, activation='sigmoid')
def call(self, inputs):
x = self.block_1(input)
return self.block_2(x)