Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

new dmlc interface

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/io/dmlc_simple.cpp
View File

@ -25,7 +25,7 @@ class SingleFileSplit : public dmlc::InputSplit {
virtual ~SingleFileSplit(void) { virtual ~SingleFileSplit(void) {
if (!use_stdin_) std::fclose(fp_); 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; if (end_of_file_) return false;
out_data->clear(); out_data->clear();
while (true) { while (true) {

2
src/io/simple_dmatrix-inl.hpp
View File

@ -94,7 +94,7 @@ class DMatrixSimple : public DataMatrix {
dmlc::InputSplit::Create(uri, rank, npart); dmlc::InputSplit::Create(uri, rank, npart);
this->Clear(); this->Clear();
std::string line; std::string line;
while (in->ReadLine(&line)) { while (in->ReadRecord(&line)) {
float label; float label;
std::istringstream ss(line); std::istringstream ss(line);
std::vector<RowBatch::Entry> feats; std::vector<RowBatch::Entry> feats;