Function GetFeature

catboost/libs/fstr/feature_str.cpp:298–328

}

TFeature GetFeature(const TFullModel& model, const TModelSplit& split) {
    TFeature result;
    result.Type = split.Type;
    switch(result.Type) {
        case ESplitType::FloatFeature:
            result.FeatureIdx = split.FloatFeature.FloatFeature;
            break;
        case ESplitType::OneHotFeature:
            result.FeatureIdx = split.OneHotFeature.CatFeatureIdx;
            break;
        case ESplitType::OnlineCtr:
            result.Ctr = split.OnlineCtr.Ctr;
            break;
        case ESplitType::EstimatedFeature:
            result.EstimatedFeature = TModelEstimatedFeature{
                split.EstimatedFeature.ModelEstimatedFeature.SourceFeatureId,
                split.EstimatedFeature.ModelEstimatedFeature.CalcerId,
                split.EstimatedFeature.ModelEstimatedFeature.LocalId,
                split.EstimatedFeature.ModelEstimatedFeature.SourceFeatureType
            };
            result.FeatureCalcerType = GetEstimatedFeatureCalcerType(
                model,
                split.EstimatedFeature.ModelEstimatedFeature
            );
            result.FeatureIdx = split.EstimatedFeature.ModelEstimatedFeature.SourceFeatureId;
            break;
        default:
            CB_ENSURE(false, "Unsupported split type " << result.Type);
    }
    return result;
}