| 133 | } |
| 134 | |
| 135 | TVector<int> GetBinFeatureCombinationClassByDepth( |
| 136 | const TModelTrees& forest, |
| 137 | const TVector<int>& binFeatureCombinationClass, |
| 138 | size_t treeIdx |
| 139 | ) { |
| 140 | const size_t depthOfTree = forest.GetModelTreeData()->GetTreeSizes()[treeIdx]; |
| 141 | TVector<int> binFeatureCombinationClassByDepth(depthOfTree); |
| 142 | for (size_t depth = 0; depth < depthOfTree; ++depth) { |
| 143 | const size_t remainingDepth = depthOfTree - depth - 1; |
| 144 | const int combinationClass = binFeatureCombinationClass[ |
| 145 | forest.GetModelTreeData()->GetTreeSplits()[forest.GetModelTreeData()->GetTreeStartOffsets()[treeIdx] + remainingDepth] |
| 146 | ]; |
| 147 | binFeatureCombinationClassByDepth[depth] = combinationClass; |
| 148 | } |
| 149 | return binFeatureCombinationClassByDepth; |
| 150 | } |
| 151 | |
| 152 | EFeatureCalcerType GetEstimatedFeatureCalcerType( |
| 153 | const TFullModel& model, |
no test coverage detected