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( include_directories(
${Platform_SOURCE_DIR}/src/Platform ${Platform_SOURCE_DIR}/src/Platform/modules
${Platform_SOURCE_DIR}/lib/PyClassifiers/src/PyClassifiers ${Platform_SOURCE_DIR}/lib/PyClassifiers/src/PyClassifiers
${Python3_INCLUDE_DIRS} ${Python3_INCLUDE_DIRS}
${Platform_SOURCE_DIR}/lib/Files ${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; float total_score = 0, total_score_train = 0, score_train, score_test;
folding::Fold* fold; folding::Fold* fold;
double nodes = 0.0;
if (stratified) if (stratified)
fold = new folding::StratifiedKFold(nFolds, y, seed); fold = new folding::StratifiedKFold(nFolds, y, seed);
else else
@@ -217,6 +218,8 @@ int main(int argc, char** argv)
auto [Xtrain, ytrain] = extract_indices(train, Xd, y); auto [Xtrain, ytrain] = extract_indices(train, Xd, y);
auto [Xtest, ytest] = extract_indices(test, Xd, y); auto [Xtest, ytest] = extract_indices(test, Xd, y);
clf->fit(Xtrain, ytrain, features, className, states); clf->fit(Xtrain, ytrain, features, className, states);
std::cout << "Nodes: " << clf->getNumberOfNodes() << std::endl;
nodes += clf->getNumberOfNodes();
score_train = clf->score(Xtrain, ytrain); score_train = clf->score(Xtrain, ytrain);
score_test = clf->score(Xtest, ytest); 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 << "Score Test : " << score_test << std::endl;
std::cout << "-------------------------------------------------------------------------------" << std::endl; std::cout << "-------------------------------------------------------------------------------" << std::endl;
} }
std::cout << "Nodes: " << nodes / nFolds << std::endl;
std::cout << "**********************************************************************************" << std::endl; std::cout << "**********************************************************************************" << std::endl;
std::cout << "Average Score Train: " << total_score_train / nFolds << 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; std::cout << "Average Score Test : " << total_score / nFolds << std::endl;return 0;