rmontanana/Platform
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add node count to sample

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-02-06 09:48:09 +01:00
parent 5993ece4fd
commit b490d406a2
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sample/CMakeLists.txt
View File

@@ -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

4
sample/sample.cc
View File

@@ -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;