| 793 | }; |
| 794 | |
| 795 | TTreeCalcFunction GetCalcTreesFunction( |
| 796 | const TModelTrees& trees, |
| 797 | size_t docCountInBlock, |
| 798 | bool calcIndexesOnly |
| 799 | ) { |
| 800 | const bool areTreesOblivious = trees.IsOblivious(); |
| 801 | const bool isSingleDoc = (docCountInBlock == 1); |
| 802 | const bool isSingleClassModel = (trees.GetDimensionsCount() == 1); |
| 803 | const bool needXorMask = !trees.GetOneHotFeatures().empty(); |
| 804 | return FunctorTemplateParamsSubstitutor<CalcTreeFunctionInstantiationGetter>::Call( |
| 805 | areTreesOblivious, isSingleDoc, isSingleClassModel, needXorMask, calcIndexesOnly); |
| 806 | } |
| 807 | } |
no test coverage detected