Update BoostAODE class structure

This commit is contained in:
Ricardo Montañana Gómez 2024-04-02 09:52:40 +02:00
parent a5a29eb66f
commit 56b5158ff3
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
2 changed files with 12 additions and 11 deletions

View File

@ -11,16 +11,7 @@
#include "bayesnet/utils/loguru.cpp"
namespace bayesnet {
struct {
std::string CFS = "CFS";
std::string FCBF = "FCBF";
std::string IWSS = "IWSS";
}SelectFeatures;
struct {
std::string ASC = "asc";
std::string DESC = "desc";
std::string RAND = "rand";
}Orders;
BoostAODE::BoostAODE(bool predict_voting) : Ensemble(predict_voting)
{
validHyperparameters = {

View File

@ -5,6 +5,16 @@
#include "bayesnet/feature_selection/FeatureSelect.h"
#include "Ensemble.h"
namespace bayesnet {
struct {
std::string CFS = "CFS";
std::string FCBF = "FCBF";
std::string IWSS = "IWSS";
}SelectFeatures;
struct {
std::string ASC = "asc";
std::string DESC = "desc";
std::string RAND = "rand";
}Orders;
class BoostAODE : public Ensemble {
public:
BoostAODE(bool predict_voting = false);
@ -23,7 +33,7 @@ namespace bayesnet {
std::string order_algorithm; // order to process the KBest features asc, desc, rand
bool convergence = false; //if true, stop when the model does not improve
bool selectFeatures = false; // if true, use feature selection
std::string select_features_algorithm = "desc"; // Selected feature selection algorithm
std::string select_features_algorithm = Orders.DESC; // Selected feature selection algorithm
FeatureSelect* featureSelector = nullptr;
double threshold = -1;
};