Implement block update algorithm fix in BoostAODE

2024-04-11 00:02:43 +02:00
parent cf9b5716ac
commit a2de1c9522
2 changed files with 66 additions and 9 deletions
--- a/tests/TestBoostAODE.cc
+++ b/tests/TestBoostAODE.cc
@@ -157,13 +157,18 @@ TEST_CASE("Bisection", "[BoostAODE]")
 TEST_CASE("Block Update", "[BoostAODE]")
 {
    auto clf = bayesnet::BoostAODE();
-    auto raw = RawDatasets("mfeat-factors", true);
+    // auto raw = RawDatasets("mfeat-factors", true);
+    auto raw = RawDatasets("glass", 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);