(self, data_size=10, perceptron_size=100, num_classes=10)
| 80 | class MLP(model.Model): |
| 81 | |
| 82 | def __init__(self, data_size=10, perceptron_size=100, num_classes=10): |
| 83 | super(MLP, self).__init__() |
| 84 | self.num_classes = num_classes |
| 85 | self.dimension = 2 |
| 86 | |
| 87 | self.relu = layer.ReLU() |
| 88 | self.linear1 = layer.Linear(perceptron_size) |
| 89 | self.linear2 = layer.Linear(num_classes) |
| 90 | self.softmax_cross_entropy = layer.SoftMaxCrossEntropy() |
| 91 | |
| 92 | def forward(self, inputs): |
| 93 | y = self.linear1(inputs) |