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

Function GetFeatureEffectForLinearIndices

catboost/libs/fstr/calc_fstr.cpp:281–299  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

279
280
281TVector<double> GetFeatureEffectForLinearIndices(
282 const TVector<std::pair<double, TFeature>>& featureEffect,
283 const TFullModel& model)
284{
285 TVector<TFeatureEffect> regularEffect = CalcRegularFeatureEffect(featureEffect, model);
286
287 const NCB::TFeaturesLayout layout = MakeFeaturesLayout(model);
288
289 TVector<double> effect(layout.GetExternalFeatureCount());
290 for (const auto& featureEffect : regularEffect) {
291 int externalFeatureIdx = layout.GetExternalFeatureIdx(
292 featureEffect.Feature.Index,
293 featureEffect.Feature.Type
294 );
295 effect[externalFeatureIdx] = featureEffect.Score;
296 }
297
298 return effect;
299}
300
301TVector<double> CalcRegularFeatureEffect(
302 const TFullModel& model,

Calls 4

CalcRegularFeatureEffectFunction · 0.85
GetExternalFeatureIdxMethod · 0.80
MakeFeaturesLayoutFunction · 0.50

Tested by

no test coverage detected