Skip to content
Snippets Groups Projects
Commit 6abf6dc9 authored by a.croix's avatar a.croix
Browse files

Add some logs

parent 79cfec61
No related branches found
No related tags found
1 merge request!4Neural network
Pipeline #2294 failed
...@@ -59,9 +59,10 @@ public final class MainTest { ...@@ -59,9 +59,10 @@ public final class MainTest {
NeuralNetwork nn = new NeuralNetwork(nn_parameters); NeuralNetwork nn = new NeuralNetwork(nn_parameters);
TrainingDataset dataset = new TrainingDataset(data, expected); TrainingDataset dataset = new TrainingDataset(data, expected);
List<TrainingDataset> folds = dataset.prepareFolds(3); List<TrainingDataset> folds = dataset.prepareFolds(10);
System.out.println("Neural Network learning");
HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10); HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10);
System.out.println("Wowa training");
HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10); HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10);
...@@ -72,12 +73,12 @@ public final class MainTest { ...@@ -72,12 +73,12 @@ public final class MainTest {
} }
System.out.println("Average AUC for Neural Network learning : " System.out.println("Average AUC for Neural Network learning : "
+ nn_score / 3); + nn_score / 10);
for (Double d : map_wt.values()) { for (Double d : map_wt.values()) {
wt_score = wt_score + d; wt_score = wt_score + d;
} }
System.out.println("Average AUC for WOWA learning : " + wt_score / 3); System.out.println("Average AUC for WOWA learning : " + wt_score / 10);
} }
} }
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment