Logo ROOT   6.08/07
Reference Guide
correlationsMultiClass.cxx
Go to the documentation of this file.
2 #include "TH2.h"
3 #include "TPaletteAxis.h"
4 #include "TMVA/Config.h"
5 
6 
7 // this macro plots the correlation matrix of the various input
8 // variables used in TMVA (e.g. running TMVAnalysis.C). Signal and
9 // Background are plotted separately
10 
11 // input: - Input file (result from TMVA),
12 // - use of colors or grey scale
13 // - use of TMVA plotting TStyle
14 void TMVA::correlationsMultiClass(TString dataset, TString fin , Bool_t /* isRegression */ ,
15  Bool_t /* greyScale */ , Bool_t useTMVAStyle )
16 {
17 
18  // set style and remove existing canvas'
19  TMVAGlob::Initialize( useTMVAStyle );
20 
21  // checks if file with name "fin" is already open, and if not opens one
22  TFile* file = TMVAGlob::OpenFile( fin );
23  TDirectory* vardir = (TDirectory*)file->GetDirectory(dataset.Data())->Get( "InputVariables_Id" );
24  std::vector<TString> classnames(TMVAGlob::GetClassNames(vardir));
25  // signal and background or regression problem
26  Int_t ncls = classnames.end() - classnames.begin();
27  std::vector<TString> hnames(classnames);
28 
29  const Int_t width = 600;
30  for (Int_t ic=0; ic<ncls; ic++) {
31  hnames[ic] = TString("CorrelationMatrix")+ hnames[ic];
32  TH2* h2 = (TH2*)file->GetDirectory(dataset.Data())->Get( hnames[ic] );
33  cout << "Looking for histo " << hnames[ic] << " in " << fin << endl;
34  if(!h2) {
35  cout << "Did not find histogram " << hnames[ic] << " in " << fin << endl;
36  continue;
37  }
38 
39  TCanvas* c = new TCanvas( hnames[ic],
40  Form("Correlations between MVA input variables (%s)",
41  classnames[ic].Data()),
42  ic*(width+5)+200, 0, width, width );
43  Float_t newMargin1 = 0.13;
44  Float_t newMargin2 = 0.15;
45  if (gConfig().fVariablePlotting.fUsePaperStyle) newMargin2 = 0.13;
46 
47  c->SetGrid();
48  c->SetTicks();
49  c->SetLeftMargin ( newMargin2 );
50  c->SetBottomMargin( newMargin2 );
51  c->SetRightMargin ( newMargin1 );
52  c->SetTopMargin ( newMargin1 );
53  gStyle->SetPalette( 1, 0 );
54 
55 
56  gStyle->SetPaintTextFormat( "3g" );
57 
58  h2->SetMarkerSize( 1.5 );
59  h2->SetMarkerColor( 0 );
60  Float_t labelSize = 0.040;
61  h2->GetXaxis()->SetLabelSize( labelSize );
62  h2->GetYaxis()->SetLabelSize( labelSize );
63  h2->LabelsOption( "d" );
64  h2->SetLabelOffset( 0.011 );// label offset on x axis
65 
66  h2->Draw("colz"); // color pads
67  c->Update();
68 
69  // modify properties of paletteAxis
70  TPaletteAxis* paletteAxis = (TPaletteAxis*)h2->GetListOfFunctions()->FindObject( "palette" );
71  paletteAxis->SetLabelSize( 0.03 );
72  paletteAxis->SetX1NDC( paletteAxis->GetX1NDC() + 0.02 );
73 
74  h2->Draw("textsame"); // add text
75 
76  // add comment
77  TText* t = new TText( 0.53, 0.88, "Linear correlation coefficients in %" );
78  t->SetNDC();
79  t->SetTextSize( 0.026 );
80  t->AppendPad();
81 
82  // TMVAGlob::plot_logo( );
83  c->Update();
84 
85  TString fname = dataset+"/plots/";
86  fname += hnames[ic];
87  TMVAGlob::imgconv( c, fname );
88  }
89 }
Config & gConfig()
Definition: Config.cxx:43
void imgconv(TCanvas *c, const TString &fname)
Definition: tmvaglob.cxx:212
virtual void LabelsOption(Option_t *option="h", Option_t *axis="X")
Set option(s) to draw axis with labels.
Definition: TH1.cxx:4934
float Float_t
Definition: RtypesCore.h:53
return c
R__EXTERN TStyle * gStyle
Definition: TStyle.h:418
TFile * OpenFile(const TString &fin)
Definition: tmvaglob.cxx:192
std::vector< TString > GetClassNames(TDirectory *dir)
Definition: tmvaglob.cxx:461
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format...
Definition: TFile.h:50
Basic string class.
Definition: TString.h:137
int Int_t
Definition: RtypesCore.h:41
bool Bool_t
Definition: RtypesCore.h:59
virtual void SetTopMargin(Float_t topmargin)
Set Pad top margin in fraction of the pad height.
Definition: TAttPad.cxx:130
virtual TObject * FindObject(const char *name) const
Find an object in this list using its name.
Definition: TList.cxx:497
virtual void AppendPad(Option_t *option="")
Append graphics object to current pad.
Definition: TObject.cxx:165
virtual void SetGrid(Int_t valuex=1, Int_t valuey=1)
Definition: TPad.h:318
virtual void SetMarkerColor(Color_t mcolor=1)
Set the marker color.
Definition: TAttMarker.h:43
std::vector< std::vector< double > > Data
Base class for several text objects.
Definition: TText.h:33
virtual void SetNDC(Bool_t isNDC=kTRUE)
Set NDC mode on if isNDC = kTRUE, off otherwise.
Definition: TText.cxx:813
Double_t GetX1NDC() const
Definition: TPave.h:60
virtual void SetBottomMargin(Float_t bottommargin)
Set Pad bottom margin in fraction of the pad height.
Definition: TAttPad.cxx:100
Service class for 2-Dim histogram classes.
Definition: TH2.h:36
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
virtual void SetX1NDC(Double_t x1)
Definition: TPave.h:79
char * Form(const char *fmt,...)
TAxis * GetYaxis()
Definition: TH1.h:325
virtual TDirectory * GetDirectory(const char *apath, Bool_t printError=false, const char *funcname="GetDirectory")
Find a directory named "apath".
virtual void SetLabelSize(Float_t size=0.04)
Set size of axis labels The size is expressed in per cent of the pad width.
Definition: TAttAxis.cxx:204
virtual void SetMarkerSize(Size_t msize=1)
Set the marker size.
Definition: TAttMarker.h:46
The Canvas class.
Definition: TCanvas.h:41
void correlationsMultiClass(TString dataset, TString fin="TMVA.root", Bool_t isRegression=kFALSE, Bool_t greyScale=kFALSE, Bool_t useTMVAStyle=kTRUE)
Describe directory structure in memory.
Definition: TDirectory.h:44
virtual void SetTicks(Int_t valuex=1, Int_t valuey=1)
Definition: TPad.h:338
virtual void SetRightMargin(Float_t rightmargin)
Set Pad right margin in fraction of the pad width.
Definition: TAttPad.cxx:120
Definition: file.py:1
void SetPaintTextFormat(const char *format="g")
Definition: TStyle.h:376
virtual void SetTextSize(Float_t tsize=1)
Set the text size.
Definition: TAttText.h:52
virtual void Update()
Update canvas pad buffers.
Definition: TCanvas.cxx:2183
TList * GetListOfFunctions() const
Definition: TH1.h:248
The palette painting class.
Definition: TPaletteAxis.h:33
void SetPalette(Int_t ncolors=kBird, Int_t *colors=0, Float_t alpha=1.)
See TColor::SetPalette.
Definition: TStyle.cxx:1450
TAxis * GetXaxis()
Definition: TH1.h:324
virtual void SetLabelSize(Float_t labelsize)
Definition: TPaletteAxis.h:66
virtual void SetLabelOffset(Float_t offset=0.005, Option_t *axis="X")
Set offset between axis and axis&#39; labels.
Definition: Haxis.cxx:267
virtual void SetLeftMargin(Float_t leftmargin)
Set Pad left margin in fraction of the pad width.
Definition: TAttPad.cxx:110
const char * Data() const
Definition: TString.h:349