| 173 | |
| 174 | |
| 175 | def example(): |
| 176 | |
| 177 | x = np.random.randn(10,10) |
| 178 | y = np.asarray([[0.8,0.4],[0.4,0.3],[0.34,0.45],[0.67,0.32], |
| 179 | [0.88,0.67],[0.78,0.77],[0.55,0.66],[0.55,0.43],[0.54,0.1], |
| 180 | [0.1,0.5]]) |
| 181 | |
| 182 | model = BPNN() |
| 183 | model.add_layer(DenseLayer(10)) |
| 184 | model.add_layer(DenseLayer(20)) |
| 185 | model.add_layer(DenseLayer(30)) |
| 186 | model.add_layer(DenseLayer(2)) |
| 187 | |
| 188 | model.build() |
| 189 | |
| 190 | model.summary() |
| 191 | |
| 192 | model.train(xdata=x,ydata=y,train_round=100,accuracy=0.01) |
| 193 | |
| 194 | if __name__ == '__main__': |
| 195 | example() |