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

Execution time added

parent 6abf6dc9
No related branches found
No related tags found
1 merge request!4Neural network
Pipeline #2295 failed
...@@ -60,11 +60,18 @@ public final class MainTest { ...@@ -60,11 +60,18 @@ public final class MainTest {
TrainingDataset dataset = new TrainingDataset(data, expected); TrainingDataset dataset = new TrainingDataset(data, expected);
List<TrainingDataset> folds = dataset.prepareFolds(10); List<TrainingDataset> folds = dataset.prepareFolds(10);
System.out.println("Neural Network learning");
HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10); long start_time = System.currentTimeMillis();
System.out.println("Wowa training"); System.out.println("Wowa training");
HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10); HashMap<AbstractSolution, Double> map_wt = trainer.runKFold(folds, 10);
long end_time = System.currentTimeMillis();
System.out.println("Execution time : " + (end_time - start_time) / 1000 + " seconds");
start_time = System.currentTimeMillis();
System.out.println("Neural Network learning");
HashMap<MultiLayerNetwork, Double> map_nn = nn.runKFold(folds, 10);
end_time = System.currentTimeMillis()
System.out.println("Execution time : " + (end_time - start_time) / 1000 + " seconds");
double nn_score = 0.0; double nn_score = 0.0;
double wt_score = 0.0; double wt_score = 0.0;
......
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