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

minor fix, add openmp

Browse Source
This commit is contained in:
Tianqi Chen 2014-08-26 20:02:10 -07:00
parent 9d2c1cf9f5
commit 3e5cb25830
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -123,7 +123,7 @@ struct EvalAMS : public IEvaluator {
const double br = 10.0; const double br = 10.0;
unsigned thresindex = 0; unsigned thresindex = 0;
double s_tp = 0.0, b_fp = 0.0, tams = 0.0; double s_tp = 0.0, b_fp = 0.0, tams = 0.0;
for (unsigned i = 0; i < ndata-1 && i < ntop; ++i) { for (unsigned i = 0; i < static_cast<unsigned>(ndata-1) && i < ntop; ++i) {
const unsigned ridx = rec[i].second; const unsigned ridx = rec[i].second;
const float wt = info.weights[ridx]; const float wt = info.weights[ridx];
if (info.labels[ridx] > 0.5f) { if (info.labels[ridx] > 0.5f) {

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

@ -169,7 +169,7 @@ class SoftmaxMultiClassObj : public IObjFunction {
std::vector<bst_gpair> &gpair = *out_gpair; std::vector<bst_gpair> &gpair = *out_gpair;
gpair.resize(preds.size()); gpair.resize(preds.size());
const unsigned nstep = static_cast<unsigned>(info.labels.size() * nclass); const unsigned nstep = static_cast<unsigned>(info.labels.size() * nclass);
const unsigned ndata = static_cast<bst_omp_uint>(preds.size() / nclass); const bst_omp_uint ndata = static_cast<bst_omp_uint>(preds.size() / nclass);
#pragma omp parallel #pragma omp parallel
{ {
std::vector<float> rec(nclass); std::vector<float> rec(nclass);

1
windows/xgboost/xgboost.vcxproj
View File

@ -99,6 +99,7 @@
<Optimization>MaxSpeed</Optimization> <Optimization>MaxSpeed</Optimization>
<FunctionLevelLinking>true</FunctionLevelLinking> <FunctionLevelLinking>true</FunctionLevelLinking>
<IntrinsicFunctions>true</IntrinsicFunctions> <IntrinsicFunctions>true</IntrinsicFunctions>
<OpenMPSupport>true</OpenMPSupport>
</ClCompile> </ClCompile>
<Link> <Link>
<GenerateDebugInformation>true</GenerateDebugInformation> <GenerateDebugInformation>true</GenerateDebugInformation>