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

Function GetCalcTreesFunction

catboost/libs/model/cpu/evaluator_impl.cpp:795–806  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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}

Callers 4

CalcGenericFunction · 0.85
CalcMethod · 0.85
CalcLeafIndexesMethod · 0.85
CalcLeafIndexesGenericFunction · 0.85

Calls 5

CallFunction · 0.50
IsObliviousMethod · 0.45
GetDimensionsCountMethod · 0.45
emptyMethod · 0.45
GetOneHotFeaturesMethod · 0.45

Tested by

no test coverage detected