TrainMVA.C File Reference

#include "TFile.h"
#include "TTree.h"
#include "TMVA/Factory.h"
#include <string>
#include <vector>
#include <map>
#include <iostream>
#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 101 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 66 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();
}