Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- fc_net = NeuralNetwork()
- (fc_net
- .add(FCLayer(100))
- .add(ActivationLayer(tanh, tanh_prime))
- .add(FCLayer(50))
- .add(ActivationLayer(tanh, tanh_prime))
- .add(FCLayer(10))
- .add(ActivationLayer(tanh, tanh_prime))
- .add(ActivationLayer(softmax, softmax_prime))
- .use(mse, mse_prime)
- .fit(x_train[0:1000], y_train[0:1000], epochs=30, learning_rate=0.1)
- )
- preds = net.predict(x_test[:100])
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement