Method WriteModel

catboost/libs/model/model_export/cpp_exporter.cpp:157–252

    }

    void TCatboostModelToCppConverter::WriteModel(bool forCatFeatures, const TFullModel& model, const THashMap<ui32, TString>* catFeaturesHashToString) {
        TIndent indent(0);
        TSequenceCommaSeparator comma;

        Out << "/* Model data */" << '\n';

        int binaryFeatureCount = forCatFeatures
                ? model.ModelTrees->GetEffectiveBinaryFeaturesBucketsCount()
                : GetBinaryFeatureCount(model);

        Out << indent++ << "static const struct CatboostModel {" << '\n';
        Out << indent << "CatboostModel() = default;" << '\n';
        Out << indent << "unsigned int FloatFeatureCount = " << model.GetNumFloatFeatures() << ";" << '\n';
        Out << indent << "unsigned int CatFeatureCount = " << model.GetNumCatFeatures() << ";" << '\n';
        Out << indent << "unsigned int BinaryFeatureCount = " << binaryFeatureCount << ";" << '\n';
        Out << indent << "unsigned int TreeCount = " << model.GetTreeCount() << ";" << '\n';

        Out << indent++ << "std::vector<std::vector<float>> FloatFeatureBorders = {" << '\n';
        comma.ResetCount(model.ModelTrees->GetFloatFeatures().size());
        for (const auto& floatFeature : model.ModelTrees->GetFloatFeatures()) {
            Out << indent << "{"
                << OutputArrayInitializer([&floatFeature](size_t i) { return FloatToString(floatFeature.Borders[i], PREC_NDIGITS, 9); }, floatFeature.Borders.size())
                << "}" << comma << '\n';
        }
        Out << --indent << "};" << '\n';

        Out << indent << "unsigned int TreeDepth[" << model.ModelTrees->GetModelTreeData()->GetTreeSizes().size() << "] = {" << OutputArrayInitializer(model.ModelTrees->GetModelTreeData()->GetTreeSizes()) << "};" << '\n';
        Out << indent << "unsigned int TreeSplits[" << model.ModelTrees->GetModelTreeData()->GetTreeSplits().size() << "] = {" << OutputArrayInitializer(model.ModelTrees->GetModelTreeData()->GetTreeSplits()) << "};" << '\n';

        if (forCatFeatures) {
            const auto& bins = model.ModelTrees->GetRepackedBins();
            Out << indent << "unsigned char TreeSplitIdxs[" << bins.size() << "] = {" << OutputArrayInitializer([&bins](size_t i) { return (int)bins[i].SplitIdx; }, bins.size()) << "};" << '\n';
            Out << indent << "unsigned short TreeSplitFeatureIndex[" << bins.size() << "] = {" << OutputArrayInitializer([&bins](size_t i) { return (int)bins[i].FeatureIndex; }, bins.size()) << "};" << '\n';
            Out << indent << "unsigned char TreeSplitXorMask[" << bins.size() << "] = {" << OutputArrayInitializer([&bins](size_t i) { return (int)bins[i].XorMask; }, bins.size()) << "};" << '\n';

            Out << indent << "unsigned int CatFeaturesIndex[" << model.ModelTrees->GetCatFeatures().size() << "] = {"
                << OutputArrayInitializer([&model](size_t i) { return model.ModelTrees->GetCatFeatures()[i].Position.Index; }, model.ModelTrees->GetCatFeatures().size()) << "};" << '\n';

            Out << indent << "std::vector<unsigned int> OneHotCatFeatureIndex = {"
                << OutputArrayInitializer([&model](size_t i) { return model.ModelTrees->GetOneHotFeatures()[i].CatFeatureIndex; }, model.ModelTrees->GetOneHotFeatures().size())
                << "};" << '\n';

            Out << indent++ << "std::vector<std::vector<int>> OneHotHashValues = {" << '\n';
            comma.ResetCount(model.ModelTrees->GetOneHotFeatures().size());
            for (const auto& oneHotFeature : model.ModelTrees->GetOneHotFeatures()) {
                Out << indent << "{"
                    << OutputArrayInitializer([&oneHotFeature](size_t i) { return oneHotFeature.Values[i]; }, oneHotFeature.Values.size())
                    << "}" << comma << '\n';
            }
            Out << --indent << "};" << '\n';

            Out << indent++ << "std::vector<std::vector<float>> CtrFeatureBorders = {" << '\n';
            comma.ResetCount(model.ModelTrees->GetCtrFeatures().size());
            for (const auto& ctrFeature : model.ModelTrees->GetCtrFeatures()) {
                Out << indent << "{"
                    << OutputArrayInitializer([&ctrFeature](size_t i) { return FloatToString(ctrFeature.Borders[i], PREC_NDIGITS, 9) + "f"; }, ctrFeature.Borders.size())
                    << "}" << comma << '\n';
            }