Naaturel/ANN-framework
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Minor changes

Browse Source
This commit is contained in:
Laurent
2026-04-02 09:21:38 +02:00
parent 4c1eaff238
commit 5a73337687
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/naaturel/ANN/Main.java
View File

@@ -26,7 +26,7 @@ public class Main {
DataSet dataset = new DatasetExtractor() DataSet dataset = new DatasetExtractor()
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/table_4_12.csv", nbrClass); .extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/table_4_12.csv", nbrClass);
int[] neuronPerLayer = new int[]{100, 100, 50, dataset.getNbrLabels()}; int[] neuronPerLayer = new int[]{50, 50, 25, dataset.getNbrLabels()};
int nbrInput = dataset.getNbrInputs(); int nbrInput = dataset.getNbrInputs();
FullyConnectedNetwork network = createNetwork(neuronPerLayer, nbrInput); FullyConnectedNetwork network = createNetwork(neuronPerLayer, nbrInput);
@@ -34,7 +34,7 @@ public class Main {
System.out.println(network.synCount()); System.out.println(network.synCount());
Trainer trainer = new GradientBackpropagationTraining(); Trainer trainer = new GradientBackpropagationTraining();
trainer.train(0.01F, 2000, network, dataset); trainer.train(0.001F, 2000, network, dataset);
//plotGraph(dataset, network); //plotGraph(dataset, network);
} }

2
src/main/java/com/naaturel/ANN/domain/model/neuron/Neuron.java
View File

@@ -46,7 +46,7 @@ public class Neuron implements Model {
} }
public float calculateWeightedSum() { public float calculateWeightedSum() {
int count = synCount(); int count = weights.length;
float weightedSum = 0F; float weightedSum = 0F;
for (int i = 0; i < count; i++){ for (int i = 0; i < count; i++){
weightedSum += weights[i] * inputs[i]; weightedSum += weights[i] * inputs[i];