Trianta/arff-mining
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

arff: fixed missing value case on read, still diagnosing soybean issue

Browse Source
This commit is contained in:
Trianta 2024-04-08 17:43:07 -05:00
parent c2bba75e5a
commit db4ef4740c
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/arff/arff.cpp
View File

@ -90,6 +90,7 @@ namespace ARFF {
for (AttributeType type : attributeList) {
std::cout << type.attribute << " (" << type.values.size() << "):";
for (std::string value : type.values) {
if (value == "?") { continue; }
std::cout << " " << value;
}
std::cout << '\n';
@ -125,8 +126,10 @@ namespace ARFF {
parser >> token;
attributeList.emplace_back(token);
LogInfo("ARFF/Attribute", "Added attribute: " + token);
while (parser >> token) {
while (std::getline(parser, token, ',')) {
// Clean token from outside pieces
token.erase(std::remove(token.begin(), token.end(), ' '), token.end());
token.erase(std::remove(token.begin(), token.end(), '\t'), token.end());
token.erase(std::remove(token.begin(), token.end(), '{'), token.end());
token.erase(std::remove(token.begin(), token.end(), '}'), token.end());
token.erase(std::remove(token.begin(), token.end(), ','), token.end());
@ -135,6 +138,8 @@ namespace ARFF {
attributeList.back().AddValue(token);
LogInfo("ARFF/Attribute", "Added value: " + token);
}
// Additional missing value case
attributeList.back().AddValue("?");
}
// Add data to runtime database
@ -147,6 +152,7 @@ namespace ARFF {
LogInfo("ARFF/Data", "Added id: " + std::to_string(database.back().id));
for (int i = 0; i < attributeList.size(); ++i) {
std::getline(parser, token, ',');
token.erase(std::remove(token.begin(), token.end(), ' '), token.end());
token.erase(std::remove(token.begin(), token.end(), '\r'), token.end());
token.erase(std::remove(token.begin(), token.end(), '\n'), token.end());
database.back().values.at(i) = token;