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

expose fmatrixs

Browse Source
This commit is contained in:
tqchen 2014-04-18 18:18:19 -07:00
parent 24696071a8
commit 17559a90f9
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
booster/xgboost_data.h
View File

@ -228,8 +228,8 @@ namespace xgboost{
* \return the row id added line * \return the row id added line
*/ */
inline size_t AddRow(const std::vector<bst_uint> &findex, inline size_t AddRow(const std::vector<bst_uint> &findex,
const std::vector<bst_float> &fvalue, const std::vector<bst_float> &fvalue,
unsigned fstart = 0, unsigned fend = UINT_MAX){ unsigned fstart = 0, unsigned fend = UINT_MAX){
utils::Assert(findex.size() == fvalue.size()); utils::Assert(findex.size() == fvalue.size());
unsigned cnt = 0; unsigned cnt = 0;
for (size_t i = 0; i < findex.size(); i++){ for (size_t i = 0; i < findex.size(); i++){
@ -378,7 +378,7 @@ namespace xgboost{
utils::Assert(fi.Read(&data[0], data.size() * sizeof(REntry)), "Load FMatrixS"); utils::Assert(fi.Read(&data[0], data.size() * sizeof(REntry)), "Load FMatrixS");
} }
} }
protected: public:
/*! \brief row pointer of CSR sparse storage */ /*! \brief row pointer of CSR sparse storage */
std::vector<size_t> row_ptr_; std::vector<size_t> row_ptr_;
/*! \brief data in the row */ /*! \brief data in the row */