Function UpdateClassLabelsFromBaselineFile

catboost/libs/data/baseline.cpp:158–177

    }

    void UpdateClassLabelsFromBaselineFile(
        const TPathWithScheme& baselineFilePath,
        TVector<NJson::TJsonValue>* classLabels
    ) {
        if (baselineFilePath.Inited()) {
            CB_ENSURE_INTERNAL(classLabels != nullptr, "ClassLabels has not been specified");
            THolder<IBaselineReader> reader = GetProcessor<IBaselineReader, TBaselineReaderArgs>(
                baselineFilePath, TBaselineReaderArgs{baselineFilePath, {}}
            );
            TVector<TString> classNamesFromBaselineFile = reader->GetClassNames();
            if (classLabels->empty()) {
                classLabels->assign(classNamesFromBaselineFile.begin(), classNamesFromBaselineFile.end());
            } else {
                CB_ENSURE(
                    NCB::ClassLabelsToStrings(*classLabels) == classNamesFromBaselineFile,
                    "Inconsistent class names in baseline file"
                );
            }
        }
    }
}