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

windows changes

Browse Source
This commit is contained in:
Tianqi Chen
2015-01-18 22:54:01 -08:00
parent 7780bc45c2
commit f22ee7cb61
6 changed files with 21 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -313,7 +313,9 @@ struct EvalAuc : public IEvaluator {
}
}
if (distributed) {
float dat[2]; dat[0] = sum_auc; dat[1] = ngroup;
float dat[2];
dat[0] = static_cast<float>(sum_auc);
dat[1] = static_cast<float>(ngroup);
// approximately estimate auc using mean
rabit::Allreduce<rabit::op::Sum>(dat, 2);
return dat[0] / dat[1];
@@ -357,7 +359,9 @@ struct EvalRankList : public IEvaluator {
}
}
if (distributed) {
float dat[2]; dat[0] = sum_metric; dat[1] = ngroup;
float dat[2];
dat[0] = static_cast<float>(sum_metric);
dat[1] = static_cast<float>(ngroup);
// approximately estimate auc using mean
rabit::Allreduce<rabit::op::Sum>(dat, 2);
return dat[0] / dat[1];