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

Reworked synapses data structure

Browse Source
This commit is contained in:
Laurent
2026-04-01 22:48:06 +02:00
parent 4441b149f9
commit 5ddf6dc580
13 changed files with 77 additions and 94 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()
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/table_4_12.csv", nbrClass);
int[] neuronPerLayer = new int[]{10, 5, 10, dataset.getNbrLabels()};
int[] neuronPerLayer = new int[]{100, 100, 50, dataset.getNbrLabels()};
int nbrInput = dataset.getNbrInputs();
FullyConnectedNetwork network = createNetwork(neuronPerLayer, nbrInput);
@@ -34,7 +34,7 @@ public class Main {
System.out.println(network.synCount());
Trainer trainer = new GradientBackpropagationTraining();
trainer.train(0.01F, 2000, network, dataset);
trainer.train(0.001F, 2000, network, dataset);
plotGraph(dataset, network);
}