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

Squashed 'subtree/rabit/' changes from b15f6cd..e08542c

Browse Source 
e08542c fix doc
e95c962 remove I prefix from interface, serializable now takes in pointer

git-subtree-dir: subtree/rabit
git-subtree-split: e08542c635
This commit is contained in:
tqchen
2015-04-08 17:39:45 -07:00
parent 89244b4aec
commit 3d11f56880
29 changed files with 225 additions and 228 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
test/lazy_recover.cc
View File

@@ -8,17 +8,17 @@
using namespace rabit;
// dummy model
class Model : public rabit::ISerializable {
class Model : public rabit::Serializable {
public:
// iterations
std::vector<float> data;
// load from stream
virtual void Load(rabit::IStream &fi) {
fi.Read(&data);
virtual void Load(rabit::Stream *fi) {
fi->Read(&data);
}
/*! \brief save the model to the stream */
virtual void Save(rabit::IStream &fo) const {
fo.Write(data);
virtual void Save(rabit::Stream *fo) const {
fo->Write(data);
}
virtual void InitModel(size_t n) {
data.clear();

10
test/local_recover.cc
View File

@@ -9,17 +9,17 @@
using namespace rabit;
// dummy model
class Model : public rabit::ISerializable {
class Model : public rabit::Serializable {
public:
// iterations
std::vector<float> data;
// load from stream
virtual void Load(rabit::IStream &fi) {
fi.Read(&data);
virtual void Load(rabit::Stream *fi) {
fi->Read(&data);
}
/*! \brief save the model to the stream */
virtual void Save(rabit::IStream &fo) const {
fo.Write(data);
virtual void Save(rabit::Stream *fo) const {
fo->Write(data);
}
virtual void InitModel(size_t n, float v) {
data.clear();

10
test/model_recover.cc
View File

@@ -8,17 +8,17 @@
using namespace rabit;
// dummy model
class Model : public rabit::ISerializable {
class Model : public rabit::Serializable {
public:
// iterations
std::vector<float> data;
// load from stream
virtual void Load(rabit::IStream &fi) {
fi.Read(&data);
virtual void Load(rabit::Stream *fi) {
fi->Read(&data);
}
/*! \brief save the model to the stream */
virtual void Save(rabit::IStream &fo) const {
fo.Write(data);
virtual void Save(rabit::Stream *fo) const {
fo->Write(data);
}
virtual void InitModel(size_t n) {
data.clear();