getTPStraining.cc File Reference

#include <cstdlib>
#include <cassert>
#include <iostream>
#include <vector>
#include <filesystem>
#include <algorithm>
#include "Core/Objects/interface/Jet.h"
#include "Core/Objects/interface/Event.h"
#include "Core/CommonTools/interface/binnings.h"
#include "Core/CommonTools/interface/DASOptions.h"
#include "Core/CommonTools/interface/toolbox.h"
#include <TString.h>
#include <TFile.h>
#include <TH1.h>
#include "Math/VectorUtil.h"
#include <darwin.h>
#include "TChain.h"
#include "TTree.h"
#include "TObjString.h"
#include "TSystem.h"
#include "TROOT.h"
#include "TMVA/Factory.h"
#include "TMVA/DataLoader.h"
#include "TMVA/Tools.h"
#include "TMVA/TMVAGui.h"
#include "Core/MachineLearning/interface/TPSVariables.h"

Include dependency graph for getTPStraining.cc:

Namespaces
	DAS
	DAS::TPS

Enumerations
enum	TrainingMode { ThreeClasses = 0, TwoClasses = 1 }

Functions
void	getTPSTrain (const vector< fs::path > &inputs_TPS, const vector< fs::path > &inputs_DPS1, const vector< fs::path > &inputs_DPS2, const vector< fs::path > &inputs_SPS, const fs::path &output, const pt::ptree &config, const int steering)
int	main (int argc, char *argv[])

Function Documentation

main()

int main	(	int	argc,
		char *	argv[]
	)

{
    try {
        DT::StandardInit();
 
        vector<fs::path> inputs_TPS;
        vector<fs::path> inputs_DPS1;
        vector<fs::path> inputs_DPS2;
        vector<fs::path> inputs_SPS;
        fs::path output;
 
        auto options = DAS::Options("TPS training with selectable mode (Multi-class or Classification)",
                            DT::config);
        options.inputs("inputs_TPS",  &inputs_TPS,  "input ROOT file(s) or directory for TPS" )
               .inputs("inputs_DPS1", &inputs_DPS1, "input ROOT file(s) or directory for DPS1")
               .inputs("inputs_DPS2", &inputs_DPS2, "input ROOT file(s) or directory for DPS2")
               .inputs("inputs_SPS",  &inputs_SPS,  "input ROOT file(s) or directory for SPS" )
               .output("output", &output, "output ROOT file")
               .arg<int>("mode", "training.mode", "training mode: 0 = TPSvsDPSvsSPS, 1 = TPSvs(DPS1+DPS2+SPS)")
               .arg<int>("nThreads", "nThreads", "number of threads"); //\todo https://gitlab.cern.ch/Proto/Darwin/-/issues/108
        const auto& config = options(argc, argv);
        const int steering = options.steering();
        ROOT::EnableImplicitMT(config.get<int>("nThreads"));
 
        DAS::TPS::getTPSTrain(inputs_TPS, inputs_DPS1, inputs_DPS2, inputs_SPS, output, config, steering);
    }
    catch (boost::exception& e) {
        DE::Diagnostic(e);
        return EXIT_FAILURE;
    }
 
    return EXIT_SUCCESS;
}