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

add simple text loader

Browse Source
This commit is contained in:
tqchen 2014-03-04 11:33:33 -08:00 committed by tqchen
parent cba130c40c
commit ffcfb12515
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
booster/xgboost_data.h
View File

@ -305,6 +305,27 @@ namespace xgboost{
FMatrixS::LoadBinary( fi, col_ptr_, col_data_ );
}
}
/*!
* \brief load from text file
* \param fi input file pointer
*/
inline void LoadText( FILE *fi ){
this->Clear();
int ninst;
while( fscanf( fi, "%d", &ninst ) == 1 ){
std::vector<booster::bst_uint> findex;
std::vector<booster::bst_float> fvalue;
while( -- ninst ){
unsigned index; float value;
utils::Assert( fscanf( fi, "%u:%f", &index, &value ) == 2 );
findex.push_back( index ); fvalue.push_back( value );
}
this->AddRow( findex, fvalue );
}
// initialize column support as well
this->InitData();
fclose(fi);
}
private:
/*!
* \brief save data to binary stream