Method InitTransformedData

catboost/libs/fstr/independent_tree_shap.cpp:533–558

}

void TIndependentTreeShapParams::InitTransformedData(
    const TFullModel& model,
    const TDataProvider& dataset,
    const NCatboostOptions::TLossDescription& metricDescription,
    NPar::ILocalExecutor* localExecutor
) {
    switch (ModelOutputType)
    {
        case EExplainableModelOutput::Probability: {
            TransformedTargetOfDataset =
                ApplyModelMulti(model, *dataset.ObjectsData, EPredictionType::Probability, 0, 0, localExecutor, dataset.RawTargetData.GetBaseline());
            break;
        }
        case EExplainableModelOutput::LossFunction : {
            Metric = CreateMetricFromDescription(metricDescription, model.GetDimensionsCount()).front().Release();
            TransformedTargetOfDataset = GetTransformedTarget(
                ApproxOfDataset,
                TargetOfDataset,
                *Metric
            );
            break;
        }
        default:
            CB_ENSURE_INTERNAL(false, "Unexpected model output type for transforming data");
    }
}

TIndependentTreeShapParams::TIndependentTreeShapParams(
    const TFullModel& model,