diff --git a/tests/TestBayesModels.cc b/tests/TestBayesModels.cc index 49ed457..cfc91c4 100644 --- a/tests/TestBayesModels.cc +++ b/tests/TestBayesModels.cc @@ -138,4 +138,15 @@ TEST_CASE("Get num features & num edges", "[BayesNet]") clf.fit(raw.Xv, raw.yv, raw.featuresv, raw.classNamev, raw.statesv); REQUIRE(clf.getNumberOfNodes() == 5); REQUIRE(clf.getNumberOfEdges() == 8); +} +TEST_CASE("BoostAODE feature_select CFS") +{ + auto raw = RawDatasets("glass", true); + auto clf = bayesnet::BoostAODE(); + clf.setHyperparameters({ {"select_features", "CFS"} }); + clf.fit(raw.Xv, raw.yv, raw.featuresv, raw.classNamev, raw.statesv); + REQUIRE(clf.getNumberOfNodes() == 90); + REQUIRE(clf.getNumberOfEdges() == 153); + REQUIRE(clf.getNotes().size() == 1); + REQUIRE(clf.getNotes()[0] == "Used features in initialization: 6 of 9 with CFS"); } \ No newline at end of file