Logo ROOT   6.10/09
Reference Guide
BoostControlPlots.cxx
Go to the documentation of this file.
2 #include <vector>
3 #include <string>
4 #include "TLegend.h"
5 #include "TText.h"
6 
7 
8 
9 // input: - Input file (result from TMVA),
10 // - use of TMVA plotting TStyle
11 // this macro is based on BDTControlPlots.C
12 void TMVA::BoostControlPlots(TString dataset, TString fin , Bool_t useTMVAStyle )
13 {
14  // set style and remove existing canvas'
15  TMVAGlob::Initialize( useTMVAStyle );
16 
17  // checks if file with name "fin" is already open, and if not opens one
18  TFile* file = TMVAGlob::OpenFile( fin );
19 
20  // get all titles of the method Boost
21  TList titles;
22  TString dirname="Method_Boost";
23  UInt_t ninst = TMVA::TMVAGlob::GetListOfTitles(dirname,titles,file->GetDirectory(dataset.Data()));
24  if (ninst==0) {
25  cout << "Could not locate directory 'Method_Boost' in file " << fin << endl;
26  return;
27  }
28  // loop over all titles
29  TIter keyIter(&titles);
30  TDirectory *boostdir;
31  TKey *key;
32  while ((key = TMVAGlob::NextKey(keyIter,"TDirectory"))) {
33  boostdir = (TDirectory *)key->ReadObj();
34  boostcontrolplots(dataset, boostdir );
35  }
36 }
37 
38 void TMVA::boostcontrolplots(TString dataset, TDirectory *boostdir ) {
39 
40  const Int_t nPlots = 6;
41 
42  Int_t width = 900;
43  Int_t height = 900;
44  char cn[100];
45  const TString titName = boostdir->GetName();
46  sprintf( cn, "cv_%s", titName.Data() );
47  TCanvas *c = new TCanvas( cn, Form( "%s Control Plots", titName.Data() ),
48  width, height );
49  c->Divide(2,4);
50 
51 
52  //TString hname[nPlots]={"Booster_BoostWeight","Booster_MethodWeight","Booster_ErrFraction","Booster_OrigErrFraction"};
53 
54  TString hname[nPlots]={"BoostWeight","MethodWeight","ErrFraction","SoverBtotal","SeparationGain", "SeparationGain"};
55 
56  // Note: the ROCIntegral plots are only filled for option "Boost_DetailedMonitoring=ture" currently not filled...
57  // TString hname[nPlots]={"BoostWeight","MethodWeight","ErrFraction","ROCIntegral_test"};
58 
59  for (Int_t i=0; i<nPlots; i++){
60  Int_t color = 4;
61  TH1 *h = (TH1*) boostdir->Get(hname[i]);
62  TString plotname = h->GetName();
63  h->SetMaximum(h->GetMaximum()*1.3);
64  h->SetMinimum( 0 );
65  h->SetMarkerColor(color);
66  h->SetMarkerSize( 0.7 );
67  h->SetMarkerStyle( 24 );
68  h->SetLineWidth(2);
69  h->SetLineColor(color);
70  h->Draw();
71  c->Update();
72  }
73 
74  // draw combined ROC plots
75 
76  TString hname_roctest[2] ={"ROCIntegral_test", "ROCIntegralBoosted_test"};
77  TString hname_roctrain[2]={"ROCIntegral_train", "ROCIntegralBoosted_train"};
78  TString htitle[2] = {"ROC integral of single classifier", "ROC integral of boosted method"};
79 
80  for (Int_t i=0; i<2; i++){
81  Int_t color = 4;
82  TPad * cPad = (TPad*)c->cd(nPlots+i+1);
83  TH1 *htest = (TH1*) boostdir->Get(hname_roctest[i]);
84  TH1 *htrain = (TH1*) boostdir->Get(hname_roctrain[i]);
85 
86  // check if filled
87  // Bool_t histFilled = (htest->GetMaximum() > 0 || htrain->GetMaximum() > 0);
88  Bool_t histFilled = (htest && htrain);
89 
90  if (!htest) htest = new TH1F("htest","",2,0,1);
91  if (!htrain) htrain = new TH1F("htrain","",2,0,1);
92 
93  htest->SetTitle(htitle[i]);
94  htest->SetMaximum(1.0);
95  htest->SetMinimum(0.0);
96  htest->SetMarkerColor(color);
97  htest->SetMarkerSize( 0.7 );
98  htest->SetMarkerStyle( 24 );
99  htest->SetLineWidth(2);
100  htest->SetLineColor(color);
101  htest->Draw();
102  htrain->SetMaximum(1.0);
103  htrain->SetMinimum(0.0);
104  htrain->SetMarkerColor(color-2);
105  htrain->SetMarkerSize( 0.7 );
106  htrain->SetMarkerStyle( 24 );
107  htrain->SetLineWidth(2);
108  htrain->SetLineColor(color-2);
109  htrain->Draw("same");
110 
111  if (histFilled) {
112  TLegend *legend= new TLegend( cPad->GetLeftMargin(),
113  0.2 + cPad->GetBottomMargin(),
114  cPad->GetLeftMargin() + 0.6,
115  cPad->GetBottomMargin() );
116  legend->AddEntry(htest, TString("testing sample"), "L");
117  legend->AddEntry(htrain, TString("training sample (orig. weights)"), "L");
118  legend->SetFillStyle( 1 );
119  legend->SetBorderSize(1);
120  legend->SetMargin( 0.3 );
121  legend->Draw("same");
122  }
123  else {
124  TText* t = new TText();
125  t->SetTextSize( 0.056 );
126  t->SetTextColor( 2 );
127  t->DrawTextNDC( .2, 0.6, "Use MethodBoost option: \"Boost_DetailedMonitoring\" " );
128  t->DrawTextNDC( .2, 0.51, "to fill this histograms" );
129  }
130 
131  c->Update();
132  }
133 
134  // write to file
135  TString fname = dataset+Form( "/plots/%s_ControlPlots", titName.Data() );
136  TMVAGlob::imgconv( c, fname );
137 
138 }
139 
140 
virtual const char * GetName() const
Returns name of object.
Definition: TNamed.h:47
virtual void SetLineWidth(Width_t lwidth)
Set the line width.
Definition: TAttLine.h:43
virtual Double_t GetMaximum(Double_t maxval=FLT_MAX) const
Return maximum value smaller than maxval of bins in the range, unless the value has been overridden b...
Definition: TH1.cxx:7666
void imgconv(TCanvas *c, const TString &fname)
Definition: tmvaglob.cxx:212
Float_t GetLeftMargin() const
Definition: TAttPad.h:44
virtual void SetMaximum(Double_t maximum=-1111)
Definition: TH1.h:375
This class displays a legend box (TPaveText) containing several legend entries.
Definition: TLegend.h:23
virtual TObject * Get(const char *namecycle)
Return pointer to object identified by namecycle.
Definition: TDirectory.cxx:729
Float_t GetBottomMargin() const
Definition: TAttPad.h:43
TFile * OpenFile(const TString &fin)
Definition: tmvaglob.cxx:192
virtual void Draw(Option_t *option="")
Draw this legend with its current attributes.
Definition: TLegend.cxx:452
THist< 1, float, THistStatContent, THistStatUncertainty > TH1F
Definition: THist.hxx:311
TH1 * h
Definition: legend2.C:5
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format...
Definition: TFile.h:46
virtual void SetMinimum(Double_t minimum=-1111)
Definition: TH1.h:376
Basic string class.
Definition: TString.h:129
int Int_t
Definition: RtypesCore.h:41
bool Bool_t
Definition: RtypesCore.h:59
void SetMargin(Float_t margin)
Definition: TLegend.h:69
virtual void SetFillStyle(Style_t fstyle)
Set the fill area style.
Definition: TAttFill.h:39
TKey * NextKey(TIter &keyIter, TString className)
Definition: tmvaglob.cxx:357
TLegend * legend
Definition: pirndm.C:35
TVirtualPad * cd(Int_t subpadnumber=0)
Set Current pad.
Definition: TPad.cxx:565
virtual TText * DrawTextNDC(Double_t x, Double_t y, const char *text)
Draw this text with new coordinates in NDC.
Definition: TText.cxx:202
void BoostControlPlots(TString dataset, TString fin="TMVA.root", Bool_t useTMVAStyle=kTRUE)
virtual void SetMarkerColor(Color_t mcolor=1)
Set the marker color.
Definition: TAttMarker.h:38
Base class for several text objects.
Definition: TText.h:23
Book space in a file, create I/O buffers, to fill them, (un)compress them.
Definition: TKey.h:24
A doubly linked list.
Definition: TList.h:43
virtual void SetLineColor(Color_t lcolor)
Set the line color.
Definition: TAttLine.h:40
void Initialize(Bool_t useTMVAStyle=kTRUE)
Definition: tmvaglob.cxx:176
virtual void Draw(Option_t *option="")
Draw this histogram with options.
Definition: TH1.cxx:2851
unsigned int UInt_t
Definition: RtypesCore.h:42
The most important graphics class in the ROOT system.
Definition: TPad.h:29
char * Form(const char *fmt,...)
virtual void SetMarkerStyle(Style_t mstyle=1)
Set the marker style.
Definition: TAttMarker.h:40
virtual TDirectory * GetDirectory(const char *apath, Bool_t printError=false, const char *funcname="GetDirectory")
Find a directory named "apath".
virtual void SetMarkerSize(Size_t msize=1)
Set the marker size.
Definition: TAttMarker.h:41
The Canvas class.
Definition: TCanvas.h:31
TLegendEntry * AddEntry(const TObject *obj, const char *label="", Option_t *option="lpf")
Add a new entry to this legend.
Definition: TLegend.cxx:359
Describe directory structure in memory.
Definition: TDirectory.h:34
void boostcontrolplots(TString dataset, TDirectory *boostdir)
The TH1 histogram class.
Definition: TH1.h:56
virtual TObject * ReadObj()
To read a TObject* from the file.
Definition: TKey.cxx:730
Definition: file.py:1
virtual void SetTextColor(Color_t tcolor=1)
Set the text color.
Definition: TAttText.h:43
virtual void SetTitle(const char *title)
Change (i.e.
Definition: TH1.cxx:6028
virtual void SetTextSize(Float_t tsize=1)
Set the text size.
Definition: TAttText.h:46
virtual void SetBorderSize(Int_t bordersize=4)
Definition: TPave.h:70
const char * Data() const
Definition: TString.h:347
UInt_t GetListOfTitles(TDirectory *rfdir, TList &titles)
Definition: tmvaglob.cxx:635