Function mode_select_features

catboost/app/mode_select_features.cpp:126–200

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

    #if defined(USE_MPI)
    char** args = const_cast<char**>(argv);
        auto& mpiManager = NCudaLib::GetMpiManager();
    mpiManager.Start(&argc, &args);
        CATBOOST_DEBUG_LOG << (mpiManager.IsMaster() ? "MASTER" : "SLAVE") << Endl;
    if (!mpiManager.IsMaster()) {
        CATBOOST_DEBUG_LOG << "Running MPI slave" << Endl;
        RunSlave();
        return 0;
    }
    #endif

    TCatBoostOptions catBoostOptions(ETaskType::CPU /* no matter */);
    TOutputFilesOptions outputOptions;
    TPoolLoadParams poolLoadParams;
    TFeaturesSelectOptions featuresSelectOptions;
    LoadOptions(
        argc,
        argv,
        &catBoostOptions,
        &outputOptions,
        &poolLoadParams,
        &featuresSelectOptions
    );

    TSetLogging inThisScope(catBoostOptions.LoggingLevel);

    auto trainerEnv = NCB::CreateTrainerEnv(catBoostOptions);

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

    CATBOOST_INFO_LOG << "Loading pools..." << Endl;
    auto pools = LoadPools(
        catBoostOptions,
        poolLoadParams,
        &catBoostOptions.DataProcessingOptions->HasTimeFlag,
        &executor
    );

    TVector<TEvalResult> evalResults(pools.Test.size());

    const TFeaturesSelectionSummary summary = SelectFeatures(
        catBoostOptions,
        outputOptions,
        &poolLoadParams,
        featuresSelectOptions,
        /*evalMetricDescriptor*/ Nothing(),
        pools,
        /*dstModel*/ nullptr,
        /*evalResults*/ GetMutablePointers(evalResults),
        /*metricsAndTimeHistory*/ nullptr,
        &executor
    );
    SaveSummaryToFile(summary, featuresSelectOptions.ResultPath);