From b490d406a2f4d9fc38ea73e2897c21bf19694d57 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana=20G=C3=B3mez?=
 <rmontanana@gmail.com>
Date: Tue, 6 Feb 2024 09:48:09 +0100
Subject: [PATCH] Add node count to sample

---
 sample/CMakeLists.txt | 2 +-
 sample/sample.cc      | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/sample/CMakeLists.txt b/sample/CMakeLists.txt
index f8ccf8d..f062ca4 100644
--- a/sample/CMakeLists.txt
+++ b/sample/CMakeLists.txt
@@ -1,5 +1,5 @@
 include_directories(
-    ${Platform_SOURCE_DIR}/src/Platform
+    ${Platform_SOURCE_DIR}/src/Platform/modules
     ${Platform_SOURCE_DIR}/lib/PyClassifiers/src/PyClassifiers
     ${Python3_INCLUDE_DIRS}
     ${Platform_SOURCE_DIR}/lib/Files
diff --git a/sample/sample.cc b/sample/sample.cc
index b96436f..b3f95a6 100644
--- a/sample/sample.cc
+++ b/sample/sample.cc
@@ -195,6 +195,7 @@ int main(int argc, char** argv)
     }
     float total_score = 0, total_score_train = 0, score_train, score_test;
     folding::Fold* fold;
+    double nodes = 0.0;
     if (stratified)
         fold = new folding::StratifiedKFold(nFolds, y, seed);
     else
@@ -217,6 +218,8 @@ int main(int argc, char** argv)
             auto [Xtrain, ytrain] = extract_indices(train, Xd, y);
             auto [Xtest, ytest] = extract_indices(test, Xd, y);
             clf->fit(Xtrain, ytrain, features, className, states);
+            std::cout << "Nodes: " << clf->getNumberOfNodes() << std::endl;
+            nodes += clf->getNumberOfNodes();
             score_train = clf->score(Xtrain, ytrain);
             score_test = clf->score(Xtest, ytest);
         }
@@ -230,6 +233,7 @@ int main(int argc, char** argv)
         std::cout << "Score Test : " << score_test << std::endl;
         std::cout << "-------------------------------------------------------------------------------" << std::endl;
     }
+    std::cout << "Nodes: " << nodes / nFolds << std::endl;
     std::cout << "**********************************************************************************" << std::endl;
     std::cout << "Average Score Train: " << total_score_train / nFolds << std::endl;
     std::cout << "Average Score Test : " << total_score / nFolds << std::endl;return 0;