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

big change, change interface to template, everything still OK

Browse Source
This commit is contained in:
tqchen
2014-03-03 22:16:37 -08:00
committed by tqchen
parent fad6522a53
commit 9da9861377
8 changed files with 135 additions and 114 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
regression/xgboost_reg.h
View File

@@ -174,7 +174,7 @@ namespace xgboost{
#pragma omp parallel for schedule( static )
for( unsigned j = 0; j < ndata; ++ j ){
preds[j] = mparam.PredTransform
( mparam.base_score + base_model.Predict( data.data[j], -1 ) );
( mparam.base_score + base_model.Predict( data.data, j, -1 ) );
}
}
private:
@@ -186,7 +186,7 @@ namespace xgboost{
#pragma omp parallel for schedule( static )
for( unsigned j = 0; j < ndata; ++ j ){
preds[j] = mparam.PredTransform
( mparam.base_score + base_model.Predict( data.data[j], buffer_offset + j ) );
( mparam.base_score + base_model.Predict( data.data, j, buffer_offset + j ) );
}
}