Add dump_cpt classifier test

This commit is contained in:
2024-04-11 18:16:06 +02:00
parent 09a1369122
commit 464fe029ea
2 changed files with 9 additions and 1 deletions

View File

@@ -79,6 +79,14 @@ TEST_CASE("Topological order", "[Classifier]")
REQUIRE(order[2] == "sepalwidth");
REQUIRE(order[3] == "petalwidth");
}
TEST_CASE("Dump_cpt", "[Classifier]")
{
auto model = bayesnet::TAN();
auto raw = RawDatasets("iris", true);
model.fit(raw.Xt, raw.yt, raw.featurest, raw.classNamet, raw.statest);
auto cpt = model.dump_cpt();
REQUIRE(cpt.size() == 1713);
}
TEST_CASE("Not fitted model", "[Classifier]")
{
auto model = bayesnet::TAN();