new dmlc interface

This commit is contained in:
tqchen 2015-04-11 20:28:50 -07:00
parent f55f8f023f
commit 24207d96fe
2 changed files with 2 additions and 2 deletions

View File

@ -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) {

View File

@ -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<RowBatch::Entry> feats;