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

fix by giulio

Browse Source
This commit is contained in:
tqchen@graphlab.com 2014-08-25 12:10:45 -07:00
parent c78a2164c2
commit 6da62159d0
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/data.h
View File

@ -373,8 +373,8 @@ class FMatrixS : public FMatrixInterface<FMatrixS>{
unsigned ncol = static_cast<unsigned>(this->NumCol()); unsigned ncol = static_cast<unsigned>(this->NumCol());
#pragma omp parallel for schedule(static) #pragma omp parallel for schedule(static)
for (unsigned i = 0; i < ncol; ++i) { for (unsigned i = 0; i < ncol; ++i) {
std::sort(&col_data_[col_ptr_[i]], std::sort(&col_data_[0] + col_ptr_[i],
&col_data_[col_ptr_[i + 1]], Entry::CmpValue); &col_data_[0] + col_ptr_[i + 1], Entry::CmpValue);
} }
} }

2
src/tree/model.h
View File

@ -259,10 +259,12 @@ class TreeModel {
} }
/*! \brief get leaf vector given nid */ /*! \brief get leaf vector given nid */
inline bst_float* leafvec(int nid) { inline bst_float* leafvec(int nid) {
if (leaf_vector.size() == 0) return NULL;
return &leaf_vector[nid * param.size_leaf_vector]; return &leaf_vector[nid * param.size_leaf_vector];
} }
/*! \brief get leaf vector given nid */ /*! \brief get leaf vector given nid */
inline const bst_float* leafvec(int nid) const{ inline const bst_float* leafvec(int nid) const{
if (leaf_vector.size() == 0) return NULL;
return &leaf_vector[nid * param.size_leaf_vector]; return &leaf_vector[nid * param.size_leaf_vector];
} }
/*! \brief initialize the model */ /*! \brief initialize the model */