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

[PLUGIN] Add densify parser

Browse Source
This commit is contained in:
tqchen
2016-01-25 11:56:16 -08:00
parent 88e362732f
commit b27b51f60e
5 changed files with 90 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/data/data.cc
View File

@@ -181,7 +181,7 @@ DMatrix* DMatrix::Load(const std::string& uri,
std::string ftype = file_format;
if (file_format == "auto") ftype = "libsvm";
std::unique_ptr<dmlc::Parser<uint32_t> > parser(
dmlc::Parser<uint32_t>::Create(fname.c_str(), partid, npart, ftype.c_str()));
dmlc::Parser<uint32_t>::Create(fname.c_str(), partid, npart, file_format.c_str()));
DMatrix* dmat = DMatrix::Create(parser.get(), cache_file);
if (!silent) {
LOG(CONSOLE) << dmat->info().num_row << 'x' << dmat->info().num_col << " matrix with "