TrainMVA.C File Reference

#include "TFile.h"
#include "TMVA/Factory.h"
#include "TTree.h"
#include <iostream>
#include <map>
#include <string>
#include <vector>
#include "MVAPIDResult.h"

Go to the source code of this file.

Functions
void	BuildTree (std::string inFile, std::string outFile)
void	TrainMVA (std::vector< std::string > signalFiles, std::vector< std::string > backgroundFiles, std::string outputFile, std::string jobName)
void	PrintRes (std::string inFile)

Function Documentation

void BuildTree	(	std::string	inFile,
		std::string	outFile
	)

Definition at line 12 of file TrainMVA.C.

References anab::MVAPIDResult::concentration, anab::MVAPIDResult::conicalness, anab::MVAPIDResult::coreHaloRatio, anab::MVAPIDResult::dEdxEnd, anab::MVAPIDResult::dEdxEndRatio, anab::MVAPIDResult::dEdxStart, anab::MVAPIDResult::evalRatio, anab::MVAPIDResult::isStoppingReco, anab::MVAPIDResult::isTrack, anab::MVAPIDResult::length, and anab::MVAPIDResult::nSpacePoints.

{
  TFile* fIn = new TFile(inFile.c_str());
  TTree* tr = (TTree*)fIn->Get("pid/MVAPID");
  std::vector<anab::MVAPIDResult>* mvares = 0;
  tr->SetBranchAddress("MVAResult", &mvares);
  TFile* fOut = new TFile(outFile.c_str(), "RECREATE");
  TTree* mvaTree = new TTree("mvaTree", "mvaTree");

  float evalRatio, concentration, coreHaloRatio, conicalness;
  float dEdxStart, dEdxEnd, dEdxEndRatio;
  float length;
  int isTrack, isStoppingReco;

  mvaTree->Branch("evalRatio", &evalRatio, "evalRatio/F");
  mvaTree->Branch("concentration", &concentration, "concentration/F");
  mvaTree->Branch("coreHaloRatio", &coreHaloRatio, "coreHaloRatio/F");
  mvaTree->Branch("conicalness", &conicalness, "conicalness/F");
  mvaTree->Branch("dEdxStart", &dEdxStart, "dEdxStart/F");
  mvaTree->Branch("dEdxEnd", &dEdxEnd, "dEdxEnd/F");
  mvaTree->Branch("dEdxEndRatio", &dEdxEndRatio, "dEdxEndRatio/F");
  mvaTree->Branch("length", &length, "length/F");
  mvaTree->Branch("isTrack", &isTrack, "isTrack/I");
  mvaTree->Branch("isStoppingReco", &isStoppingReco, "isStoppingReco/I");

  for (int iEntry = 0; iEntry < tr->GetEntries(); ++iEntry) {
    tr->GetEntry(iEntry);
    if (!mvares->size()) continue;

    anab::MVAPIDResult* biggestTrack = &((*mvares)[0]);
    for (unsigned int iRes = 0; iRes != mvares->size(); ++iRes) {
      if ((((*mvares)[iRes]).nSpacePoints) > (biggestTrack->nSpacePoints)) {
        biggestTrack = &((*mvares)[iRes]);
      }
    }

    evalRatio = biggestTrack->evalRatio;
    concentration = biggestTrack->concentration;
    coreHaloRatio = biggestTrack->coreHaloRatio;
    conicalness = biggestTrack->conicalness;
    dEdxStart = biggestTrack->dEdxStart;
    dEdxEnd = biggestTrack->dEdxEnd;
    dEdxEndRatio = biggestTrack->dEdxEndRatio;
    length = biggestTrack->length;
    isTrack = biggestTrack->isTrack;
    isStoppingReco = biggestTrack->isStoppingReco;

    mvaTree->Fill();
  }

  fIn->Close();
  fOut->Write();
  fOut->Close();
}

void PrintRes

(

std::string

inFile

)

Definition at line 108 of file TrainMVA.C.

References anab::MVAPIDResult::mvaOutput, and anab::MVAPIDResult::nSpacePoints.

{

  TFile* fIn = new TFile(inFile.c_str());
  TTree* tr = (TTree*)fIn->Get("pid/ANAB");
  std::vector<anab::MVAPIDResult>* mvares = 0;
  tr->SetBranchAddress("MVAResult", &mvares);

  for (int iEntry = 0; iEntry < tr->GetEntries(); ++iEntry) {
    tr->GetEntry(iEntry);
    if (!mvares->size()) continue;

    anab::MVAPIDResult* biggestTrack = &((*mvares)[0]);
    for (unsigned int iRes = 0; iRes != mvares->size(); ++iRes) {
      if ((((*mvares)[iRes]).nSpacePoints) > (biggestTrack->nSpacePoints)) {
        biggestTrack = &((*mvares)[iRes]);
      }
    }

    std::cout << biggestTrack->mvaOutput.at(string("ANN")) << std::endl;
  }

  fIn->Close();
}

void TrainMVA	(	std::vector< std::string >	signalFiles,
		std::vector< std::string >	backgroundFiles,
		std::string	outputFile,
		std::string	jobName
	)

Definition at line 67 of file TrainMVA.C.

{

  TFile* fOut = new TFile(outputFile.c_str(), "RECREATE");
  TMVA::Factory* factory = new TMVA::Factory(jobName.c_str(), fOut, "");

  std::vector<TTree*> sigTrees;

  for (std::vector<std::string>::iterator fIter = signalFiles.begin(); fIter != signalFiles.end();
       ++fIter) {
    TFile* fIn = new TFile(fIter->c_str());
    factory->AddSignalTree((TTree*)fIn->Get("mvaTree"));
  }

  for (std::vector<std::string>::iterator fIter = backgroundFiles.begin();
       fIter != backgroundFiles.end();
       ++fIter) {
    TFile* fIn = new TFile(fIter->c_str());
    factory->AddBackgroundTree((TTree*)fIn->Get("mvaTree"));
  }

  factory->AddVariable("evalRatio", 'F');
  factory->AddVariable("concentration", 'F');
  factory->AddVariable("coreHaloRatio", 'F');
  factory->AddVariable("conicalness", 'F');
  factory->AddVariable("dEdxStart", 'F');
  factory->AddVariable("dEdxEnd", 'F');
  factory->AddVariable("dEdxEndRatio", 'F');

  factory->BookMethod(TMVA::Types::kTMlpANN, "ANN", "");
  factory->BookMethod(TMVA::Types::kBDT, "BDT", "");
  factory->TrainAllMethods();
  factory->TestAllMethods();
  factory->EvaluateAllMethods();
  fOut->Write();
  fOut->Close();
}