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

isolate iserializable

Browse Source
This commit is contained in:
tqchen
2014-12-19 17:36:42 -08:00
parent 8c35cff02c
commit 6bf282c6c2
14 changed files with 150 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/mock.h
View File

@@ -33,14 +33,14 @@ public:
rabit::Allreduce<OP>(sendrecvbuf, count);
}
inline int LoadCheckPoint(utils::ISerializable *global_model,
utils::ISerializable *local_model) {
inline int LoadCheckPoint(ISerializable *global_model,
ISerializable *local_model) {
utils::Assert(verify(loadCheckpoint), "[%d] error when loading checkpoint", rank);
return rabit::LoadCheckPoint(global_model, local_model);
}
inline void CheckPoint(const utils::ISerializable *global_model,
const utils::ISerializable *local_model) {
inline void CheckPoint(const ISerializable *global_model,
const ISerializable *local_model) {
utils::Assert(verify(checkpoint), "[%d] error when checkpointing", rank);
rabit::CheckPoint(global_model, local_model);
}

6
test/test_local_recover.cpp
View File

@@ -29,16 +29,16 @@ inline void CallEnd(const char *fun, int ntrial, int iter) {
}
// dummy model
class Model : public rabit::utils::ISerializable {
class Model : public rabit::ISerializable {
public:
// iterations
std::vector<float> data;
// load from stream
virtual void Load(rabit::utils::IStream &fi) {
virtual void Load(rabit::IStream &fi) {
fi.Read(&data);
}
/*! \brief save the model to the stream */
virtual void Save(rabit::utils::IStream &fo) const {
virtual void Save(rabit::IStream &fo) const {
fo.Write(data);
}
virtual void InitModel(size_t n, float v) {

6
test/test_model_recover.cpp
View File

@@ -29,16 +29,16 @@ inline void CallEnd(const char *fun, int ntrial, int iter) {
}
// dummy model
class Model : public rabit::utils::ISerializable {
class Model : public rabit::ISerializable {
public:
// iterations
std::vector<float> data;
// load from stream
virtual void Load(rabit::utils::IStream &fi) {
virtual void Load(rabit::IStream &fi) {
fi.Read(&data);
}
/*! \brief save the model to the stream */
virtual void Save(rabit::utils::IStream &fo) const {
virtual void Save(rabit::IStream &fo) const {
fo.Write(data);
}
virtual void InitModel(size_t n) {