Function set_key

catboost/app/mode_metadata.cpp:33–70

};

int set_key(int argc, const char* argv[]) {
    TCommonMetaInfoParams params;
    TString key;
    TString value;
    TString outputModelPath;
    EModelType outputModelFormat = EModelType::CatboostBinary;
    auto parser = NLastGetopt::TOpts();
    parser.AddHelpOption();
    params.BindParser(parser);
    parser.AddLongOption("key", "key name")
        .RequiredArgument("NAME")
        .StoreResult(&key);
    parser.AddLongOption("value", "value")
        .RequiredArgument("VALUE")
        .StoreResult(&value);
    parser.AddLongOption('o', "output-model-path")
            .OptionalArgument("PATH")
            .Handler1T<TString>([&outputModelPath, &outputModelFormat](const TString& path) {
                outputModelPath = path;
                outputModelFormat = NCatboostOptions::DefineModelFormat(path);
            });
    parser.AddLongOption("output-model-format")
            .OptionalArgument("output model format")
            .Handler1T<TString>([&outputModelFormat](const TString& format) {
                outputModelFormat = FromString<EModelType>(format);
            });
    parser.SetFreeArgsMax(0);
    NLastGetopt::TOptsParseResult parserResult{&parser, argc, argv};
    params.LoadModel();
    params.Model.ModelInfo[key] = value;
    if (outputModelPath.empty()) {
        CB_ENSURE(params.ModelPath.size() == 1, "Metadata manipulation requires exactly one model");
        NCB::ExportModel(params.Model, params.ModelPath[0], outputModelFormat);
    } else {
        NCB::ExportModel(params.Model, outputModelPath, outputModelFormat);
    }
    return 0;
}

int get_keys(int argc, const char* argv[]) {
    TCommonMetaInfoParams params;