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

Implement string features

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2024-06-12 18:17:03 +02:00
parent dbefa02d9c
commit 00068c05ed
3 changed files with 48983 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
tests/TestArffFiles.cc
View File

@@ -116,4 +116,17 @@ TEST_CASE("Load with class name as first attribute", "[ArffFiles]")
for (int i = 120; i < 124; ++i)
REQUIRE(arff.getY()[i] == expected_y[i - 120]);
}
TEST_CASE("Adult dataset", "[ArffFiles]")
{
ArffFiles arff;
arff.load(Paths::datasets("adult"));
REQUIRE(arff.getClassName() == "class");
REQUIRE(arff.getClassType() == "{ <=50K, >50K}");
REQUIRE(arff.getLabels().size() == 2);
REQUIRE(arff.getLabels()[0] == "<=50K");
REQUIRE(arff.getLabels()[1] == ">50K");
REQUIRE(arff.getSize() == 32561);
REQUIRE(arff.getLines().size() == 32561);
REQUIRE(arff.getLines()[0] == "39, State-gov, 77516, Bachelors, 13, Never-married, Adm-clerical, Not-in-family, White ");
}