fazo96/AIrium
1
0
Fork 0
mirror of https://github.com/fazo96/AIrium.git synced 2025-01-10 09:34:20 +01:00
Code Issues Releases Wiki Activity

removed shitty workarounds

Browse Source
This commit is contained in:
Enrico Fasoli 2015-07-01 17:35:43 +02:00
parent 4195c08745
commit a78f27627f
2 changed files with 7 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
core/src/logic/neural/Brain.java
View File

@ -11,22 +11,22 @@ public class Brain {
public static final float bias = 0.5f;
private Neuron[][] neurons;
private int nInputs;
public Brain(int nInputs, int nOutputs, int hiddenLayers, int neuronsPerHiddenLayer) {
this.nInputs = nInputs;
neurons = new Neuron[hiddenLayers + 2][Math.max(nInputs, Math.max(nOutputs, neuronsPerHiddenLayer))];
neurons = new Neuron[hiddenLayers + 2][];
populate(nInputs, nOutputs, hiddenLayers, neuronsPerHiddenLayer);
}
private void populate(int nInputs, int nOutputs, int hiddenLayers, int neuronsPerHiddenLayer) {
// Create input neurons
neurons[0] = new Neuron[nInputs];
for (int i = 0; i < nInputs; i++) {
neurons[0][i] = new Neuron(0, bias, this);
Log.log(Log.DEBUG, "Adding Input Layer Neuron " + (i + 1));
}
// popiulate hidden layers
for (int i = 0; i < hiddenLayers; i++) {
neurons[i + 1] = new Neuron[neuronsPerHiddenLayer];
for (int j = 0; j < neuronsPerHiddenLayer; j++) {
// create neuron
Neuron n = new Neuron(i + 1, bias, this);
@ -35,6 +35,7 @@ public class Brain {
}
}
// populate output layer
neurons[hiddenLayers + 1] = new Neuron[nOutputs];
for (int i = 0; i < nOutputs; i++) {
// add neuron
Neuron n = new Neuron(hiddenLayers + 1, bias, this);
@ -100,30 +101,21 @@ public class Brain {
res[i] = new float[neurons[i].length][];
for (int j = 0; j < neurons[i].length; j++) // neurons per layer
{
if (neurons[i][j] != null) {
res[i][j] = neurons[i][j].mutate(mutationFactor);
}
}
}
return res;
}
private void clearCache() {
for (int i = 1; i < neurons.length; i++) {
for (int j = 0; j < neurons[i].length; j++) {
if (neurons[i][j] != null) {
neurons[i][j].clearCache();
}
}
}
}
public Neuron[][] getNeurons() {
return neurons;
}
public int howManyInputNeurons() {
return nInputs;
}
}

4
core/src/logic/neural/Neuron.java
View File

@ -39,10 +39,8 @@ public class Neuron {
private void scramble() {
// init weights
if (layer > 1) {
if (layer > 0) {
weights = new float[brain.getNeurons()[layer - 1].length];
} else if (layer == 1) {
weights = new float[brain.howManyInputNeurons()];
} else { // layer 0
isInputNeuron = true;
weights = new float[0];