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

initial correction for vec tree

Browse Source
This commit is contained in:
tqchen
2014-08-24 18:48:19 -07:00
parent 7874c2559b
commit c640485f1d
2 changed files with 40 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -24,9 +24,10 @@ template<typename Derived>
struct EvalEWiseBase : public IEvaluator {
virtual float Eval(const std::vector<float> &preds,
const MetaInfo &info) const {
utils::Check(preds.size() == info.labels.size(),
utils::Check(info.labels.size() != 0, "label set cannot be empty");
utils::Check(preds.size() % info.labels.size() == 0,
"label and prediction size not match");
const unsigned ndata = static_cast<unsigned>(preds.size());
const unsigned ndata = static_cast<unsigned>(info.labels.size());
float sum = 0.0, wsum = 0.0;
#pragma omp parallel for reduction(+: sum, wsum) schedule(static)
for (unsigned i = 0; i < ndata; ++i) {