Start to add test coverage
This commit is contained in:
@@ -20,8 +20,8 @@ public class Main {
|
||||
DataSet dataset = new DatasetExtractor()
|
||||
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/table_2_9.csv");
|
||||
|
||||
DataSet orDataset = new DatasetExtractor()
|
||||
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/or.csv");
|
||||
DataSet andDataset = new DatasetExtractor()
|
||||
.extract("C:/Users/Laurent/Desktop/ANN-framework/src/main/resources/assets/and.csv");
|
||||
|
||||
List<Synapse> syns = new ArrayList<>();
|
||||
syns.add(new Synapse(new Input(0), new Weight(0)));
|
||||
@@ -29,12 +29,12 @@ public class Main {
|
||||
|
||||
Bias bias = new Bias(new Weight(0));
|
||||
|
||||
Neuron neuron = new SimplePerceptron(syns, bias, new Linear());
|
||||
Neuron neuron = new SimplePerceptron(syns, bias, new Heaviside());
|
||||
Layer layer = new Layer(List.of(neuron));
|
||||
Network network = new Network(List.of(layer));
|
||||
|
||||
Trainer trainer = new GradientDescentTraining();
|
||||
trainer.train(network, dataset);
|
||||
Trainer trainer = new SimpleTraining();
|
||||
trainer.train(network, andDataset);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user