Function CalcAndOutputPredictionDiff

catboost/libs/fstr/compare_documents.cpp:264–281

}

void CalcAndOutputPredictionDiff(
    const TFullModel& model,
    const NCB::TDataProvider& dataProvider,
    const TString& outputPath,
    NPar::ILocalExecutor* localExecutor
) {
    auto factorImpact = GetPredictionDiff(model, dataProvider, localExecutor);
    TVector<std::pair<double, int>> impacts;
    for (const auto& impact: factorImpact) {
        impacts.push_back({impact[0], impacts.size()});
    }
    StableSort(impacts.begin(), impacts.end(), std::greater<std::pair<double, int>>());

    TFileOutput out(outputPath);
    for (const auto& impact: impacts) {
        out << impact.first << " " << impact.second << Endl;
    }
}