rmontanana/mdlp
1
1
Fork 0
mirror of https://github.com/rmontanana/mdlp.git synced 2025-08-16 07:55:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix attribute name extraction in ArffFiles

Browse Source
This commit is contained in:
Ricardo Montañana Gómez
2023-07-06 16:15:23 +02:00
parent 5679d607e5
commit be1917d05b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/ArffFiles.cpp
View File

@@ -63,7 +63,7 @@ void ArffFiles::load(const string& fileName, bool classLast)
type = ""; type = "";
while (ss >> type_w) while (ss >> type_w)
type += type_w + " "; type += type_w + " ";
attributes.emplace_back(attribute, trim(type)); attributes.emplace_back(trim(attribute), trim(type));
continue; continue;
} }
if (line[0] == '@') { if (line[0] == '@') {
@@ -111,8 +111,8 @@ void ArffFiles::generateDataset(bool classLast)
string ArffFiles::trim(const string& source) string ArffFiles::trim(const string& source)
{ {
string s(source); string s(source);
s.erase(0, s.find_first_not_of(" \n\r\t")); s.erase(0, s.find_first_not_of(" '\n\r\t"));
s.erase(s.find_last_not_of(" \n\r\t") + 1); s.erase(s.find_last_not_of(" '\n\r\t") + 1);
return s; return s;
} }