rmontanana/BayesNet
rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests Packages Projects Releases 3 Wiki Activity

Fix Initialize worse_model_count if model accuracy is better in BoostAODE

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2024-03-11 22:33:50 +01:00
parent 1986d05c34
commit 8f3bb47cfd
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bayesnet/ensembles/BoostAODE.cc
View File

@ -282,6 +282,12 @@ namespace bayesnet {
}
exitCondition = n_models >= maxModels && repeatSparent || worse_model_count > tolerance;
}
if (worse_model_count > tolerance) {
notes.push_back("Convergence threshold reached & last model eliminated");
significanceModels.pop_back();
models.pop_back();
n_models--;
}
if (featuresUsed.size() != features.size()) {
notes.push_back("Used features in train: " + std::to_string(featuresUsed.size()) + " of " + std::to_string(features.size()));
status = WARNING;