Skip to content

GitLab

Unverified Commit dc584070 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix bug in parser

parent fdb39237
Show whitespace changes
Inline Side-by-side
Showing with 27 additions and 24 deletions
src/io/parser.cpp
View file @ dc584070
...@@ -112,7 +112,10 @@ std::vector<std::string> ReadKLineFromFile(const char* filename, bool header, in ...@@ -112,7 +112,10 @@ std::vector<std::string> ReadKLineFromFile(const char* filename, bool header, in
for (int i = 0; i < k; ++i) { for (int i = 0; i < k; ++i) {
if (!tmp_file.eof()) { if (!tmp_file.eof()) {
GetLine(&tmp_file, &cur_line, reader.get(), &buffer, buffer_size); GetLine(&tmp_file, &cur_line, reader.get(), &buffer, buffer_size);
cur_line = Common::Trim(cur_line);
if (!cur_line.empty()) {
ret.push_back(cur_line); ret.push_back(cur_line);
}
} else { } else {
break; break;
} }
...@@ -142,7 +145,7 @@ DataType GetDataType(const std::vector<std::string>& lines, int* num_col) { ...@@ -142,7 +145,7 @@ DataType GetDataType(const std::vector<std::string>& lines, int* num_col) {
} else if (comma_cnt > 0) { } else if (comma_cnt > 0) {
type = DataType::CSV; type = DataType::CSV;
} }
} } else if (lines.size() > 1) {
int comma_cnt2 = 0; int comma_cnt2 = 0;
int tab_cnt2 = 0; int tab_cnt2 = 0;
int colon_cnt2 = 0; int colon_cnt2 = 0;
...@@ -167,7 +170,7 @@ DataType GetDataType(const std::vector<std::string>& lines, int* num_col) { ...@@ -167,7 +170,7 @@ DataType GetDataType(const std::vector<std::string>& lines, int* num_col) {
} }
} }
} }
}
if (type == DataType::LIBSVM) { if (type == DataType::LIBSVM) {
int max_col_idx = 0; int max_col_idx = 0;
for (size_t i = 0; i < lines.size(); ++i) { for (size_t i = 0; i < lines.size(); ++i) {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment