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

Conditional Entropy test

Browse Source
This commit is contained in:
Ricardo Montañana Gómez 2024-05-15 11:28:09 +02:00
parent 7efd95095c
commit ef3c74633c
Signed by: rmontanana
GPG Key ID: 46064262FD9A7ADE
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
tests/TestBayesMetrics.cc
View File

@ -100,6 +100,25 @@ TEST_CASE("Entropy Test", "[Metrics]")
REQUIRE(result == Catch::Approx(0.693147180559945).epsilon(raw.epsilon)); REQUIRE(result == Catch::Approx(0.693147180559945).epsilon(raw.epsilon));
} }
TEST_CASE("Conditional Entropy", "[Metrics]") TEST_CASE("Conditional Entropy", "[Metrics]")
{
auto raw = RawDatasets("iris", true);
bayesnet::Metrics metrics(raw.dataset, raw.features, raw.className, raw.classNumStates);
auto expected = std::map<std::pair<int, int>, double>{
{ { 0, 1 }, 1.32674 },
{ { 0, 2 }, 0.236253 },
{ { 0, 3 }, 0.1202 },
{ { 1, 2 }, 0.252551 },
{ { 1, 3 }, 0.10515 },
{ { 2, 3 }, 0.108323 },
};
for (int i = 0; i < raw.features.size() - 1; ++i) {
for (int j = i + 1; j < raw.features.size(); ++j) {
double result = metrics.conditionalEntropy(raw.dataset.index({ i, "..." }), raw.dataset.index({ j, "..." }), raw.yt, raw.weights);
REQUIRE(result == Catch::Approx(expected.at({ i, j })).epsilon(raw.epsilon));
}
}
}
TEST_CASE("Conditional Mutual Information", "[Metrics]")
{ {
auto raw = RawDatasets("iris", true); auto raw = RawDatasets("iris", true);
bayesnet::Metrics metrics(raw.dataset, raw.features, raw.className, raw.classNumStates); bayesnet::Metrics metrics(raw.dataset, raw.features, raw.className, raw.classNumStates);