diff --git a/src/io/dmlc_simple.cpp b/src/io/dmlc_simple.cpp index 9e3d50e82..4296d5caa 100644 --- a/src/io/dmlc_simple.cpp +++ b/src/io/dmlc_simple.cpp @@ -25,7 +25,7 @@ class SingleFileSplit : public dmlc::InputSplit { virtual ~SingleFileSplit(void) { if (!use_stdin_) std::fclose(fp_); } - virtual bool ReadLine(std::string *out_data) { + virtual bool ReadRecord(std::string *out_data) { if (end_of_file_) return false; out_data->clear(); while (true) { diff --git a/src/io/simple_dmatrix-inl.hpp b/src/io/simple_dmatrix-inl.hpp index d7d8d5fce..e0ea7899e 100644 --- a/src/io/simple_dmatrix-inl.hpp +++ b/src/io/simple_dmatrix-inl.hpp @@ -94,7 +94,7 @@ class DMatrixSimple : public DataMatrix { dmlc::InputSplit::Create(uri, rank, npart); this->Clear(); std::string line; - while (in->ReadLine(&line)) { + while (in->ReadRecord(&line)) { float label; std::istringstream ss(line); std::vector feats;