rmontanana/BayesNet
1
0
Fork 0
Code Issues 2 Pull Requests 1 Actions Packages Projects Releases 6 Wiki Activity

Refactor TestUtils to allow partial and shuffle dataset load

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-04-30 02:11:14 +02:00
parent ae469b8146
commit 793b2d3cd5
4 changed files with 139 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/TestBoostAODE.cc
View File

@@ -140,7 +140,7 @@ TEST_CASE("Oddities", "[BoostAODE]")
TEST_CASE("Bisection Best", "[BoostAODE]")
{
auto clf = bayesnet::BoostAODE();
auto raw = RawDatasets("mfeat-factors", true, 500);
auto raw = RawDatasets("mfeat-factors", true, 300, true);
clf.setHyperparameters({
{"bisection", true},
{"maxTolerance", 3},
@@ -149,8 +149,8 @@ TEST_CASE("Bisection Best", "[BoostAODE]")
{"convergence_best", true},
});
clf.fit(raw.X_train, raw.y_train, raw.features, raw.className, raw.states);
REQUIRE(clf.getNumberOfNodes() == 217);
REQUIRE(clf.getNumberOfEdges() == 431);
REQUIRE(clf.getNumberOfNodes() == 434);
REQUIRE(clf.getNumberOfEdges() == 862);
REQUIRE(clf.getNotes().size() == 3);
REQUIRE(clf.getNotes()[0] == "Convergence threshold reached & 15 models eliminated");
REQUIRE(clf.getNotes()[1] == "Used features in train: 16 of 216");
@@ -162,7 +162,7 @@ TEST_CASE("Bisection Best", "[BoostAODE]")
}
TEST_CASE("Bisection Best vs Last", "[BoostAODE]")
{
auto raw = RawDatasets("mfeat-factors", true, 1500);
auto raw = RawDatasets("mfeat-factors", true, 500);
auto clf = bayesnet::BoostAODE(true);
auto hyperparameters = nlohmann::json{
{"select_features", "IWSS"},