Function mode_eval_feature

catboost/app/mode_eval_feature.cpp:26–124

int mode_eval_feature(int argc, const char* argv[]) {
    ConfigureMalloc();

    NJson::TJsonValue catBoostFlatJsonOptions;
    TString paramsFile;
    NCatboostOptions::TPoolLoadParams poolLoadParams;
    NJson::TJsonValue featureEvalJsonOptions;
    ParseFeatureEvalCommandLine(
        argc,
        argv,
        &catBoostFlatJsonOptions,
        &featureEvalJsonOptions,
        &paramsFile,
        &poolLoadParams
    );

    CB_ENSURE(poolLoadParams.TestSetPaths.empty(), "Test files are not supported in feature evaluation mode");

    NJson::TJsonValue catBoostJsonOptions;
    NJson::TJsonValue outputOptionsJson;
    InitOptions(paramsFile, &catBoostJsonOptions, &outputOptionsJson);
    NCatboostOptions::LoadPoolMetaInfoOptions(poolLoadParams.PoolMetaInfoPath, &catBoostJsonOptions);

    ConvertIgnoredFeaturesFromStringToIndices(poolLoadParams, &catBoostFlatJsonOptions);
    ConvertFeaturesToEvaluateFromStringToIndices(poolLoadParams, &featureEvalJsonOptions);
    NCatboostOptions::PlainJsonToOptions(catBoostFlatJsonOptions, &catBoostJsonOptions, &outputOptionsJson);
    #if defined(HAVE_CUDA)
    THolder<TStopCudaManagerCallback> stopCudaManagerGuard;
    if (NCatboostOptions::GetTaskType(catBoostFlatJsonOptions) == ETaskType::GPU) {
        auto options = NCatboostOptions::LoadOptions(catBoostJsonOptions);
        stopCudaManagerGuard = StartCudaManager(
            NCudaLib::CreateDeviceRequestConfig(options),
            options.LoggingLevel);
    }
    #endif
    ConvertParamsToCanonicalFormat(poolLoadParams, &catBoostJsonOptions);
    CopyIgnoredFeaturesToPoolParams(catBoostJsonOptions, &poolLoadParams);

    const auto taskType = NCatboostOptions::GetTaskType(catBoostJsonOptions);
    NCatboostOptions::TCatBoostOptions catBoostOptions(taskType);
    catBoostOptions.Load(catBoostJsonOptions);
    NCatboostOptions::TFeatureEvalOptions featureEvalOptions;
    featureEvalOptions.Load(featureEvalJsonOptions);

    NPar::LocalExecutor().RunAdditionalThreads(catBoostOptions.SystemOptions->NumThreads - 1);

    TVector<NJson::TJsonValue> classLabels = catBoostOptions.DataProcessingOptions->ClassLabels;
    const auto objectsOrder = EObjectsOrder::Undefined;
    auto pools = NCB::ReadTrainDatasets(
        Nothing(), // taskType,
        poolLoadParams,
        objectsOrder,
        /*readTestData*/false,
        /*learnDatasetSubset*/ TDatasetSubset::MakeColumns(),
        /*testDatasetSubsets*/ {},
        catBoostOptions.DataProcessingOptions->ForceUnitAutoPairWeights,
        &classLabels,
        &NPar::LocalExecutor(),