MCPcopy Create free account
hub / github.com/catboost/catboost / AddTreeSize

Method AddTreeSize

catboost/libs/model/model.cpp:258–266  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

256 CastToSolidTree(*this)->TreeSplits.push_back(treeSplit);
257}
258void TModelTrees::AddTreeSize(int treeSize) {
259 auto& data = *CastToSolidTree(*this);
260 if (data.TreeStartOffsets.empty()) {
261 data.TreeStartOffsets.push_back(0);
262 } else {
263 data.TreeStartOffsets.push_back(data.TreeStartOffsets.back() + data.TreeSizes.back());
264 }
265 data.TreeSizes.push_back(treeSize);
266}
267
268void TModelTrees::AddLeafValue(double leafValue) {
269 CastToSolidTree(*this)->LeafValues.push_back(leafValue);

Callers 3

BuildMethod · 0.80
GetObliviousModelTreesFunction · 0.80

Calls 4

CastToSolidTreeFunction · 0.85
emptyMethod · 0.45
push_backMethod · 0.45
backMethod · 0.45

Tested by

no test coverage detected