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

Rename some stuff

Browse Source
This commit is contained in:
Laurent
2026-03-29 21:32:08 +02:00
parent 83526b72d4
commit 0fe309cd4e
38 changed files with 334 additions and 215 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3,13 +3,10 @@ package com.naaturel.ANN;
import com.naaturel.ANN.domain.model.neuron.Neuron;
import com.naaturel.ANN.domain.abstraction.Trainer;
import com.naaturel.ANN.implementation.multiLayers.Sigmoid;
import com.naaturel.ANN.implementation.multiLayers.TanH;
import com.naaturel.ANN.implementation.training.GradientBackpropagationTraining;
import com.naaturel.ANN.infrastructure.dataset.DataSet;
import com.naaturel.ANN.infrastructure.dataset.DatasetExtractor;
import com.naaturel.ANN.domain.model.neuron.*;
import com.naaturel.ANN.implementation.gradientDescent.Linear;
import com.naaturel.ANN.implementation.training.GradientDescentTraining;
import java.util.*;
@@ -20,7 +17,7 @@ public class Main {
int nbrInput = 2;
int nbrClass = 3;
int nbrLayers = 1;
int nbrLayers = 2;
DataSet dataset = new DatasetExtractor()
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/table_3_1.csv", nbrClass);
@@ -44,7 +41,7 @@ public class Main {
Layer layer = new Layer(neurons);
layers.add(layer);
}
Network network = new Network(layers);
FullyConnectedNetwork network = new FullyConnectedNetwork(layers);
Trainer trainer = new GradientBackpropagationTraining();
trainer.train(network, dataset);