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

check in linear model

Browse Source
This commit is contained in:
tqchen
2014-08-22 19:27:33 -07:00
parent 37b707e110
commit 2ac8cdb873
8 changed files with 287 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/learner/evaluation-inl.hpp
View File

@@ -28,7 +28,7 @@ struct EvalEWiseBase : public IEvaluator {
"label and prediction size not match");
const unsigned ndata = static_cast<unsigned>(preds.size());
float sum = 0.0, wsum = 0.0;
#pragma omp parallel for reduction(+:sum, wsum) schedule(static)
#pragma omp parallel for reduction(+: sum, wsum) schedule(static)
for (unsigned i = 0; i < ndata; ++i) {
const float wt = info.GetWeight(i);
sum += Derived::EvalRow(info.labels[i], preds[i]) * wt;

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

@@ -164,7 +164,7 @@ class BoostLearner {
inline void UpdateOneIter(int iter, const DMatrix<FMatrix> &train) {
this->PredictRaw(train, &preds_);
obj_->GetGradient(preds_, train.info, iter, &gpair_);
gbm_->DoBoost(gpair_, train.fmat, train.info.info);
gbm_->DoBoost(train.fmat, train.info.info, &gpair_);
}
/*!
* \brief evaluate the model for specific iteration