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

Adding neurons to layers is now more robust

Browse source
This commit is contained in:
lluni 2022-05-28 02:26:56 +02:00
parent 67d94efc39
commit cf75abf75c
2 changed files with 10 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
src/main/java/de/lluni/javann/Network.java
View file

@ -27,12 +27,17 @@ public class Network {
*/ */
public void addNeuron(int layer, int n) { public void addNeuron(int layer, int n) {
if (!(this.layers.get(layer) instanceof FCLayer)) { if (!(this.layers.get(layer) instanceof FCLayer)) {
System.out.println("This layer is not a de.lluni.javann.layers.BlankLayer"); System.out.println("This layer is not a FCLayer");
} else if (!(this.layers.get(layer + 2) instanceof FCLayer)) {
System.out.println("The next layer is not a de.lluni.javann.layers.BlankLayer");
} }
((FCLayer) this.layers.get(layer)).addNeuron(n); ((FCLayer) this.layers.get(layer)).addNeuron(n);
((FCLayer) this.layers.get(layer + 2)).updateInputSize(n);
for (int i = layer + 1; i < this.layers.size(); i++) {
if (this.getLayers().get(i) instanceof FCLayer) {
((FCLayer) this.layers.get(i)).updateInputSize(n);
break;
}
}
} }
/** /**

2
src/main/java/de/lluni/javann/layers/FCLayer.java
View file

@ -46,7 +46,7 @@ public class FCLayer extends Layer {
} }
for (int i = 0; i < newWeights.getNumElements(); i++) { for (int i = 0; i < newWeights.getNumElements(); i++) {
if (newWeights.get(i) == 0) { if (newWeights.get(i) == 0) {
newWeights.set(i, random.nextDouble(-1, 1)); newWeights.set(i, 0.1d * random.nextGaussian(0, 1));
} }
} }
this.weights = newWeights; this.weights = newWeights;