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

change input data structure

Browse Source
This commit is contained in:
tqchen
2014-02-26 11:51:58 -08:00
parent e58daa6d52
commit 3a4d0f28d9
9 changed files with 204 additions and 129 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
regression/xgboost_reg.h
View File

@@ -117,8 +117,7 @@ namespace xgboost{
this->GetGradient( preds, train_->labels, grad, hess );
std::vector<unsigned> root_index;
booster::FMatrixS::Image train_image( train_->data );
base_model.DoBoost(grad,hess,train_image,root_index);
base_model.DoBoost(grad,hess,train_->data,root_index);
}
/*!
* \brief evaluate the model for specific iteration

4
regression/xgboost_regdata.h
View File

@@ -132,8 +132,8 @@ namespace xgboost{
for( size_t i = 0; i < data.NumRow(); i ++ ){
booster::FMatrixS::Line sp = data[i];
for( unsigned j = 0; j < sp.len; j ++ ){
if( num_feature <= sp.findex[j] ){
num_feature = sp.findex[j] + 1;
if( num_feature <= sp[j].findex ){
num_feature = sp[j].findex + 1;
}
}
}