Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- int layers[] = {2, 1};
- NeuralNetwork n = new NeuralNetwork(layers, 2);
- float[][] inputs = {{0.0f, 0.0f, 0.0f}, {1.0f, 1.0f, 1.0f}, {0.0f, 1.0f, 0.0f}, {1.0f, 0.0f, 1.0f}, {0.0f, 1.0f, 1.0f}};
- float[][] outputs = {{0.0f}, {1.0f}, {1.0f}, {1.0f}, {1.0f}};
- for (int t = 0; t < 100000; t++) {
- for (int i = 0; i < inputs.length - 1; i++) {
- n.setInput(inputs[i]);
- n.activate();
- n.train(outputs[i]);
- if (t % 100 == 0) {
- System.out.println("Input: " + Arrays.toString(inputs[i]) + " Expected: " + Arrays.toString(outputs[i]) + " ANN output ->" + Arrays.toString(n.getOutput()));
- }
- }
- }
- n.setInput(inputs[4]);
- n.activate();
- System.out.println("MEGA TEST: " + Arrays.toString(n.getOutput()));
Add Comment
Please, Sign In to add comment