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

new change for mpi

Browse Source
This commit is contained in:
tqchen
2014-10-16 15:12:10 -07:00
parent a21df0770d
commit 0cf2dd39ea
8 changed files with 65 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/learner/learner-inl.hpp
View File

@@ -173,7 +173,7 @@ class BoostLearner {
inline void UpdateOneIter(int iter, const DMatrix &train) {
this->PredictRaw(train, &preds_);
obj_->GetGradient(preds_, train.info, iter, &gpair_);
gbm_->DoBoost(train.fmat(), train.info.info, &gpair_);
gbm_->DoBoost(train.fmat(), this->FindBufferOffset(train), train.info.info, &gpair_);
}
/*!
* \brief evaluate the model for specific iteration
@@ -335,7 +335,7 @@ class BoostLearner {
// gradient pairs
std::vector<bst_gpair> gpair_;
private:
protected:
// cache entry object that helps handle feature caching
struct CacheEntry {
const DMatrix *mat_;