lluni/JavaNN
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added additional XOR example with later added neurons

Browse source
This commit is contained in:
lluni 2022-05-23 19:10:43 +02:00
parent 8c82838c54
commit 4766ea0ad9
1 changed files with 32 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
src/main/java/ExampleXORBlankLayers.java Normal file
View file

@ -0,0 +1,32 @@
import org.ejml.simple.SimpleMatrix;
public class ExampleXORBlankLayers {
public static void main(String[] args) {
SimpleMatrix[] X_train = {new SimpleMatrix(new double[][]{{0, 0}}),
new SimpleMatrix(new double[][]{{0, 1}}),
new SimpleMatrix(new double[][]{{1, 0}}),
new SimpleMatrix(new double[][]{{1, 1}})};
SimpleMatrix[] y_train = {new SimpleMatrix(new double[][]{{0}}),
new SimpleMatrix(new double[][]{{1}}),
new SimpleMatrix(new double[][]{{1}}),
new SimpleMatrix(new double[][]{{0}})};
Network network = new Network();
network.addLayer(new BlankLayer(2));
network.addLayer(new ActivationLayer(ActivationFunctions::tanh, ActivationFunctions::tanhPrime));
network.addLayer(new BlankLayer(1));
network.addLayer(new ActivationLayer(ActivationFunctions::tanh, ActivationFunctions::tanhPrime));
network.addNeuron(0, 2);
network.use(LossFunctions::MSE, LossFunctions::MSEPrime);
network.fit(X_train, y_train, 1000, 0.1d);
SimpleMatrix[] output = network.predict(X_train);
for (SimpleMatrix entry : output) {
System.out.println("Prediction:");
for (int i = 0; i < entry.getNumElements(); i++) {
System.out.println(Math.round(entry.get(i)));
}
}
}
}