diff --git a/sample/sample.cc b/sample/sample.cc index 89902af..848f4a6 100644 --- a/sample/sample.cc +++ b/sample/sample.cc @@ -1,6 +1,6 @@ #include #include -#include "ensembles/BoostAODE.h" +#include std::vector discretizeDataset(std::vector& X, mdlp::labels_t& y) { diff --git a/src/classifiers/Proposal.cc b/src/classifiers/Proposal.cc index 58b2e73..5687eba 100644 --- a/src/classifiers/Proposal.cc +++ b/src/classifiers/Proposal.cc @@ -1,5 +1,5 @@ #include "Proposal.h" -#include "ArffFiles.h" +#include namespace bayesnet { Proposal::Proposal(torch::Tensor& dataset_, std::vector& features_, std::string& className_) : pDataset(dataset_), pFeatures(features_), pClassName(className_) {} diff --git a/src/classifiers/Proposal.h b/src/classifiers/Proposal.h index 3686db9..37d8a26 100644 --- a/src/classifiers/Proposal.h +++ b/src/classifiers/Proposal.h @@ -4,7 +4,7 @@ #include #include #include "bayesian_network/Network.h" -#include "CPPFImdlp.h" +#include #include "Classifier.h" namespace bayesnet { diff --git a/tests/TestUtils.h b/tests/TestUtils.h index 72954c0..474b3cf 100644 --- a/tests/TestUtils.h +++ b/tests/TestUtils.h @@ -5,8 +5,8 @@ #include #include #include -#include "ArffFiles.h" -#include "CPPFImdlp.h" +#include +#include bool file_exists(const std::string& name); std::pair, map> discretize(std::vector& X, mdlp::labels_t& y, std::vector features);