[GBM] remove need to explicit InitModel, rename save/load

This commit is contained in:
tqchen
2016-01-03 05:31:00 -08:00
parent 82ceb4de0a
commit 4b4b36d047
5 changed files with 50 additions and 59 deletions

View File

@@ -29,13 +29,13 @@ class TreeSyncher: public TreeUpdater {
int rank = rabit::GetRank();
if (rank == 0) {
for (size_t i = 0; i < trees.size(); ++i) {
trees[i]->SaveModel(&fs);
trees[i]->Save(&fs);
}
}
fs.Seek(0);
rabit::Broadcast(&s_model, 0);
for (size_t i = 0; i < trees.size(); ++i) {
trees[i]->LoadModel(&fs);
trees[i]->Load(&fs);
}
}
};