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

add python interface for xgboost

Browse Source
This commit is contained in:
tqchen
2014-05-03 23:04:02 -07:00
parent adc9400736
commit 6fd77cbb24
10 changed files with 8348 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
regrank/xgboost_regrank_data.h
View File

@@ -112,7 +112,9 @@ namespace xgboost{
unsigned ngptr;
if( fs.Read(&ngptr, sizeof(unsigned) ) != 0 ){
info.group_ptr.resize( ngptr );
utils::Assert( fs.Read(&info.group_ptr[0], sizeof(unsigned) * ngptr) != 0, "Load group file");
if( ngptr != 0 ){
utils::Assert( fs.Read(&info.group_ptr[0], sizeof(unsigned) * ngptr) != 0, "Load group file");
}
}
}
fs.Close();
@@ -143,7 +145,9 @@ namespace xgboost{
{// write out group ptr
unsigned ngptr = static_cast<unsigned>( info.group_ptr.size() );
fs.Write(&ngptr, sizeof(unsigned) );
fs.Write(&info.group_ptr[0], sizeof(unsigned) * ngptr);
if( ngptr != 0 ){
fs.Write(&info.group_ptr[0], sizeof(unsigned) * ngptr);
}
}
fs.Close();
if (!silent){