rmontanana/Pyclassifiers
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update tests to last version

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-03-06 18:04:34 +01:00
parent c8826fc3b4
commit 101fb02da5
3 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/CMakeLists.txt
View File

@@ -6,6 +6,9 @@ if(ENABLE_TESTING)
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/lib/Files
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/lib/mdlp
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/classifiers
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/bayesian_network
${PyClassifiers_SOURCE_DIR}/lib/BayesNet/src/utils
${Python3_INCLUDE_DIRS}
${TORCH_INCLUDE_DIRS}
${CMAKE_BINARY_DIR}/configured_files/include

3
tests/TestPythonClassifiers.cc
View File

@@ -35,11 +35,12 @@ TEST_CASE("Test Python Classifiers score", "[PyClassifiers]")
};
SECTION("Test Python Classifier " + name + " score ")
{
for (auto file_name : { "glass", "iris", "ecoli", "diabetes" }) {
for (std::string file_name : { "glass", "iris", "ecoli", "diabetes" }) {
auto raw = RawDatasets(file_name, false);
auto clf = models[name];
clf->fit(raw.Xt, raw.yt, raw.featurest, raw.classNamet, raw.statest);
auto score = clf->score(raw.Xt, raw.yt);
INFO("File: " + file_name + " Classifier: " + name + " Score: " + to_string(score));
REQUIRE(score == Catch::Approx(scores[{file_name, name}]).epsilon(raw.epsilon));
}
}