new dmlc interface
This commit is contained in:
parent
f55f8f023f
commit
24207d96fe
@ -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) {
|
||||
|
||||
@ -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;
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user