Add some more tests to 97% coverage

This commit is contained in:
2024-04-11 17:29:46 +02:00
parent 8eeaa1beee
commit 503ad687dc
8 changed files with 75 additions and 37 deletions

View File

@@ -157,18 +157,13 @@ TEST_CASE("Bisection", "[BoostAODE]")
TEST_CASE("Block Update", "[BoostAODE]")
{
auto clf = bayesnet::BoostAODE();
// auto raw = RawDatasets("mfeat-factors", true);
auto raw = RawDatasets("glass", true);
auto raw = RawDatasets("mfeat-factors", true);
clf.setHyperparameters({
{"bisection", true},
{"block_update", true},
{"maxTolerance", 3},
{"convergence", true},
});
// clf.setHyperparameters({
// {"block_update", true},
// });
clf.fit(raw.Xv, raw.yv, raw.featuresv, raw.classNamev, raw.statesv);
REQUIRE(clf.getNumberOfNodes() == 217);
REQUIRE(clf.getNumberOfEdges() == 431);