36. Model 클래스

 이번에는 모델을 표현하기 위한 Model 클래스를 새로 만든다. Model 클래스는 Layer 클래스의 기능을 이어받으며 시각화 메서드가 하나 추가된다.

from dezero import Layer
from dezero import utils
import dezero.functions as F
import dezero.layers as L

class Model(Layer):
    def plot(self,*inputs,to_file='model.png'):
        y=self.forward(*inputs)
        return utils.plot_dot_graph(y,verbose=True,to_file=to_file)

 

 이제 다음과 같은 코드를 작성할 수 있다.

 

• MLP 클래스

  class MLP(Model):
      def __init__(self, fc_output_sizes, activation=F.sigmoid):
          super().__init__()
          self.activation = activation
          self.layers = []
  
          for i, out_size in enumerate(fc_output_sizes):
              layer = L.Linear(out_size)
              setattr(self, 'l' + str(i), layer)
              self.layers.append(layer)
  
      def forward(self, x):
          for l in self.layers[:-1]:
              x = self.activation(l(x))
          return self.layers[-1](x)​