vtxRS.C

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file vtxRS.C

#include <TFile.h>
#include <TTree.h>
#include <TString.h>
#include <TChain.h>


#if not defined(__CINT__) || defined(__MAKECINT__)
// needs to be included when makecint runs (ACLIC)
#include "TMVA/Factory.h"
#include "TMVA/Tools.h"
#endif



TChain* handleFile(string name, string extension, string treename, int filecount){
  TChain *all = new TChain(treename.c_str());
  string temp;
  for (int i = 0; i < filecount; ++i)
  {

    ostringstream s;
    s<<i;
    temp = name+string(s.str())+extension;
    all->Add(temp.c_str());
  }
  return all;
}


void makeFactory(TTree* signalTree,std::string outfile,std::string factoryname)
{
  using namespace TMVA;
  TString jobname(factoryname.c_str());
  TFile *targetFile = new TFile(outfile.c_str(),"RECREATE");
  Factory *factory = new Factory(jobname,targetFile,"AnalysisType=Regression");
  factory->AddRegressionTree(signalTree,1.0);
  factory->AddVariable("track1_pt",'F');
  factory->AddVariable("track2_pt",'F');
  factory->AddVariable("track1_phi",'F');
  factory->AddVariable("track1_phi-track2_phi","d#phi","rad");
  factory->AddVariable("track1_eta",'F');
  factory->AddVariable("track1_eta-track2_eta","d#eta","rad");
  factory->AddVariable("(vtx_radius)/(track_layer+1)",'F');
  factory->AddVariable("vtx_radius","radius","cm");
  factory->AddTarget("tvtx_radius","radius","cm");

  string track_pT_cut = "track1_pt>.6&&track2_pt>.6";

  //string vtx_radius_cut = "vtx_radius>0"; //can I cut based on label?
  string tCutInitializer = track_pT_cut;
  TCut preTraingCuts(tCutInitializer.c_str());
  factory->PrepareTrainingAndTestTree(preTraingCuts,"nTrain_Regression=0:nTest_Regression=0");
  factory->BookMethod(Types::kKNN,"kNN14L","VarTransform=G:nkNN=16");
  factory->BookMethod(Types::kKNN,"kNN12L","VarTransform=G:nkNN=8");
  //factory->BookMethod( Types::kPDERS, "PDERS", "KernelEstimator=Teepee:NEventsMin=120:NEventsMax=210:VarTransform=G(_V0_,_V1_,_V3_,_V5_)");
  //factory->BookMethod(Types::kMLP,"MLP_ANN2","HiddenLayers=500,6");
  //factory->BookMethod(Types::kMLP,"MLP_ANN","HiddenLayers=5");


  factory->TrainAllMethods();
  factory->TestAllMethods();
  factory->EvaluateAllMethods();
  targetFile->Write();
  targetFile->Close();
}


int vtxRS(){
  using namespace std;
  string treePath = "/sphenix/user/vassalli/gammasample/conversionembededonlineanalysis";
  string treeExtension = ".root";
  string outname = "/direct/phenix+u/vassalli/sphenix/single/Training/condorout/vtxTrainRS.root";
  unsigned int nFiles=200;

  TChain *signalTree = handleFile(treePath,treeExtension,"vtxingTree",nFiles);
  makeFactory(signalTree,outname,"vtxFactory");
  /*  outname="cutTrainE.root";
      makeFactory(signalTree,backETree,outname,"eback");*/
}