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

change data format to include weight in binary file, add get weight to python

Browse Source
This commit is contained in:
tqchen
2014-05-15 14:37:56 -07:00
parent 54c486bcf1
commit 6aa190e10c
13 changed files with 31 additions and 1514 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
regrank/xgboost_regrank_data.h
View File

@@ -124,6 +124,14 @@ namespace xgboost{
}
}
}
{// load in weight
unsigned nwt;
if( fs.Read(&nwt, sizeof(unsigned) ) != 0 ){
utils::Assert( nwt == 0 || nwt == data.NumRow(), "invalid weight" );
info.weights.resize( nwt );
utils::Assert( fs.Read(&info.weights[0], sizeof(unsigned) * nwt) != 0, "Load weight file");
}
}
fs.Close();
if (!silent){
@@ -133,7 +141,6 @@ namespace xgboost{
printf("data contains %u groups\n", (unsigned)info.group_ptr.size()-1 );
}
}
this->TryLoadWeight(fname, silent);
return true;
}
/*!
@@ -153,7 +160,14 @@ namespace xgboost{
unsigned ngptr = static_cast<unsigned>( info.group_ptr.size() );
fs.Write(&ngptr, sizeof(unsigned) );
if( ngptr != 0 ){
fs.Write(&info.group_ptr[0], sizeof(unsigned) * ngptr);
fs.Write(&info.group_ptr[0], sizeof(unsigned) * ngptr);
}
}
{// write out weight
unsigned nwt = static_cast<unsigned>( info.weights.size() );
fs.Write( &nwt, sizeof(unsigned) );
if( nwt != 0 ){
fs.Write(&info.weights[0], sizeof(float) * nwt);
}
}
fs.Close();