Function GetEstimatedFeatureCalcerType

catboost/libs/fstr/util.cpp:152–165

}

EFeatureCalcerType GetEstimatedFeatureCalcerType(
    const TFullModel& model,
    const TModelEstimatedFeature& estimatedFeature
) {
    if (estimatedFeature.SourceFeatureType == EEstimatedSourceFeatureType::Text) {
        return model.TextProcessingCollection->GetCalcer(estimatedFeature.CalcerId)->Type();
    } else {
        CB_ENSURE_INTERNAL(
            estimatedFeature.SourceFeatureType == EEstimatedSourceFeatureType::Embedding,
            "Unexpected EEstimatedSourceFeatureType: " << estimatedFeature.SourceFeatureType
        );
        return model.EmbeddingProcessingCollection->GetCalcer(estimatedFeature.CalcerId)->Type();
    }
}