Fix CFS mistake

This commit is contained in:
2024-04-02 22:53:00 +02:00
parent e55365c41c
commit a1f9086780
4 changed files with 34 additions and 77 deletions

View File

@@ -145,6 +145,10 @@ namespace bayesnet {
}
featureSelector->fit();
auto cfsFeatures = featureSelector->getFeatures();
auto scores = featureSelector->getScores();
for (int i = 0; i < cfsFeatures.size(); ++i) {
LOG_F(INFO, "Feature: %d Score: %f", cfsFeatures[i], scores[i]);
}
for (const int& feature : cfsFeatures) {
featuresUsed.push_back(feature);
std::unique_ptr<Classifier> model = std::make_unique<SPODE>(feature);