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

Parallel model dump for trees. (#7040)

Browse Source
This commit is contained in:
Jiaming Yuan 2021-06-15 14:08:26 +08:00 committed by GitHub
parent 2567404ab6
commit 5c2d7a18c9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/gbm/gbtree_model.h
View File

@ -17,6 +17,8 @@
#include <string>
#include <vector>
#include "../common/threading_utils.h"
namespace xgboost {
class Json;
@ -107,12 +109,12 @@ struct GBTreeModel : public Model {
void SaveModel(Json* p_out) const override;
void LoadModel(Json const& p_out) override;
std::vector<std::string> DumpModel(const FeatureMap& fmap, bool with_stats,
std::vector<std::string> DumpModel(const FeatureMap &fmap, bool with_stats,
std::string format) const {
std::vector<std::string> dump;
for (const auto & tree : trees) {
dump.push_back(tree->DumpModel(fmap, with_stats, format));
}
std::vector<std::string> dump(trees.size());
common::ParallelFor(static_cast<omp_ulong>(trees.size()), [&](size_t i) {
dump[i] = trees[i]->DumpModel(fmap, with_stats, format);
});
return dump;
}
void CommitModel(std::vector<std::unique_ptr<RegTree> >&& new_trees,