From 55187ee5212c62c912727752a0f7aed08da8ee41 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Wed, 11 Dec 2024 10:05:24 +0100
Subject: [PATCH] Add time to experiment seed

---
 src/main/Experiment.cpp | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/main/Experiment.cpp b/src/main/Experiment.cpp
index 10997d0..65fced8 100644
--- a/src/main/Experiment.cpp
+++ b/src/main/Experiment.cpp
@@ -176,7 +176,7 @@ namespace platform {
         json confusion_matrices_train = json::array();
         std::vector<std::string> notes;
         std::vector<std::string> graphs;
-        Timer train_timer, test_timer;
+        Timer train_timer, test_timer, seed_timer;
         int item = 0;
         bool first_seed = true;
         //
@@ -184,6 +184,7 @@ namespace platform {
         //
         auto score = parse_score();
         for (auto seed : randomSeeds) {
+            seed_timer.start();
             if (!quiet) {
                 string prefix = " ";
                 if (!first_seed) {
@@ -274,8 +275,10 @@ namespace platform {
                     graphs.push_back(result);
                 }
             }
-            if (!quiet)
-                std::cout << "end. " << flush;
+            if (!quiet) {
+                seed_timer.stop();
+                std::cout << "end. [" << seed_timer.getDurationString() << "]" << std::endl;
+            }
             delete fold;
         }
         //