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

final minor

Browse Source
This commit is contained in:
tqchen 2015-04-25 21:19:42 -07:00
parent e03eabccda
commit 27340f95e4
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/dmlc/io.h
View File

@ -303,7 +303,7 @@ inline void Stream::Write(const std::vector<T> &vec) {
uint64_t sz = static_cast<uint64_t>(vec.size());
this->Write(&sz, sizeof(sz));
if (sz != 0) {
this->Write(&vec[0], sizeof(T) * sz);
this->Write(&vec[0], sizeof(T) * vec.size());
}
}
template<typename T>
@ -320,7 +320,7 @@ inline void Stream::Write(const std::string &str) {
uint64_t sz = static_cast<uint64_t>(str.length());
this->Write(&sz, sizeof(sz));
if (sz != 0) {
this->Write(&str[0], sizeof(char) * sz);
this->Write(&str[0], sizeof(char) * str.length());
}
}
inline bool Stream::Read(std::string *out_str) {