14 #include<TMVA/Factory.h>
19 TFile *
infile =
new TFile(
"l3bdt.root");
20 TTree *l3tree = (TTree*)infile->Get(
"l3tree");
22 cout <<
"Couldn't open \"l3bdt.root\"!" << endl;
27 TFile *
outfile =
new TFile(
"l3BDT_out.root",
"RECREATE");
28 TMVA::Factory *fac =
new TMVA::Factory(
"L3",outfile,
"");
31 TCut signalCut(
"Evisible>=4.0 && ((Npshits+Npschits)<=1)");
32 TCut backgroundCut(
"Evisible<4.0 && ((Npshits+Npschits)<=1)");
33 fac->SetInputTrees(l3tree, signalCut, backgroundCut);
36 fac->AddVariable(
"Nstart_counter",
'I');
37 fac->AddVariable(
"Ntof",
'I');
38 fac->AddVariable(
"Nbcal_points",
'I');
39 fac->AddVariable(
"Nbcal_clusters",
'I');
40 fac->AddVariable(
"Ebcal_points",
'F');
41 fac->AddVariable(
"Ebcal_clusters",
'F');
42 fac->AddVariable(
"Nfcal_clusters",
'I');
43 fac->AddVariable(
"Efcal_clusters",
'F');
44 fac->AddVariable(
"Ntrack_candidates",
'I');
45 fac->AddVariable(
"Ptot_candidates",
'F');
47 TCut preSelectCut(
"");
48 fac->PrepareTrainingAndTestTree(preSelectCut,
"");
49 fac->BookMethod(TMVA::Types::kBDT,
"BDT",
"");
51 fac->TrainAllMethods();
52 fac->TestAllMethods();
53 fac->EvaluateAllMethods();