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

Fix adult numeric features mistake

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2025-08-21 19:01:10 +02:00
parent f1a2349245
commit 1aa3b609e5
3 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
tests/TestBayesModels.cc
View File

@@ -531,6 +531,11 @@ TEST_CASE("Test Dataset Loading", "[Datasets]")
}
std::cout << "| " << dataset.yt[sample].item<int>() << std::endl;
}
auto features = dataset.features;
std::cout << "States:" << std::endl;
for (int i = 0; i < 14; i++) {
std::cout << i << " has " << dataset.states.at(features[i]).size() << " states." << std::endl;
}
dataset = RawDatasets("adult", false);
std::cout << "Dataset adult raw " << std::endl;
for (int sample = 0; sample < max_sample; sample++) {
@@ -539,4 +544,10 @@ TEST_CASE("Test Dataset Loading", "[Datasets]")
}
std::cout << "| " << dataset.yt[sample].item<int>() << std::endl;
}
std::cout << "States:" << std::endl;
for (int i = 0; i < 14; i++) {
std::cout << i << " has " << dataset.states.at(features[i]).size() << " states." << std::endl;
}
auto clf = bayesnet::TANLd();
clf.fit(dataset.Xt, dataset.yt, dataset.features, dataset.className, dataset.states, dataset.smoothing);
}