diff --git a/src/main/java/be/cylab/java/wowa/training/MainTest.java b/src/main/java/be/cylab/java/wowa/training/MainTest.java index 10cb9fc2b29a9bd5906cfe0c6dab0146edc2bbeb..d7d5ddb80afe7436894db6a23c41b921743e35b7 100644 --- a/src/main/java/be/cylab/java/wowa/training/MainTest.java +++ b/src/main/java/be/cylab/java/wowa/training/MainTest.java @@ -60,11 +60,18 @@ public final class MainTest { TrainingDataset dataset = new TrainingDataset(data, expected); 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"); 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 wt_score = 0.0;