Logo ROOT  
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
14void 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
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}
#define c(i)
Definition: RSha256.hxx:101
bool Bool_t
Definition: RtypesCore.h:63
int Int_t
Definition: RtypesCore.h:45
float Float_t
Definition: RtypesCore.h:57
Option_t Option_t width
char * Form(const char *fmt,...)
Formats a string in a circular formatting buffer.
Definition: TString.cxx:2447
R__EXTERN TStyle * gStyle
Definition: TStyle.h:414
virtual void SetLabelSize(Float_t size=0.04)
Set size of axis labels.
Definition: TAttAxis.cxx:206
virtual void SetMarkerColor(Color_t mcolor=1)
Set the marker color.
Definition: TAttMarker.h:38
virtual void SetMarkerSize(Size_t msize=1)
Set the marker size.
Definition: TAttMarker.h:41
virtual void SetTextSize(Float_t tsize=1)
Set the text size.
Definition: TAttText.h:47
The Canvas class.
Definition: TCanvas.h:23
Describe directory structure in memory.
Definition: TDirectory.h:45
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Definition: TFile.h:54
virtual void SetLabelOffset(Float_t offset=0.005, Option_t *axis="X")
Set offset between axis and axis' labels.
Definition: Haxis.cxx:267
virtual void LabelsOption(Option_t *option="h", Option_t *axis="X")
Sort bins with labels or set option(s) to draw axis with labels.
Definition: TH1.cxx:5356
TAxis * GetXaxis()
Get the behaviour adopted by the object about the statoverflows. See EStatOverflows for more informat...
Definition: TH1.h:319
TAxis * GetYaxis()
Definition: TH1.h:320
void Draw(Option_t *option="") override
Draw this histogram with options.
Definition: TH1.cxx:3070
TList * GetListOfFunctions() const
Definition: TH1.h:242
Service class for 2-D histogram classes.
Definition: TH2.h:30
TObject * FindObject(const char *name) const override
Find an object in this list using its name.
Definition: TList.cxx:578
virtual void AppendPad(Option_t *option="")
Append graphics object to current pad.
Definition: TObject.cxx:107
The palette painting class.
Definition: TPaletteAxis.h:29
virtual void SetLabelSize(Float_t labelsize)
Definition: TPaletteAxis.h:59
Double_t GetX1NDC() const
Definition: TPave.h:59
virtual void SetX1NDC(Double_t x1)
Definition: TPave.h:78
Basic string class.
Definition: TString.h:136
const char * Data() const
Definition: TString.h:369
void SetPaintTextFormat(const char *format="g")
Definition: TStyle.h:370
void SetPalette(Int_t ncolors=kBird, Int_t *colors=0, Float_t alpha=1.)
See TColor::SetPalette.
Definition: TStyle.cxx:1783
Base class for several text objects.
Definition: TText.h:22
virtual void SetNDC(Bool_t isNDC=kTRUE)
Set NDC mode on if isNDC = kTRUE, off otherwise.
Definition: TText.cxx:813
void Initialize(Bool_t useTMVAStyle=kTRUE)
Definition: tmvaglob.cxx:176
TFile * OpenFile(const TString &fin)
Definition: tmvaglob.cxx:192
std::vector< TString > GetClassNames(TDirectory *dir)
Definition: tmvaglob.cxx:469
void imgconv(TCanvas *c, const TString &fname)
Definition: tmvaglob.cxx:212
Config & gConfig()
void correlationsMultiClass(TString dataset, TString fin="TMVA.root", Bool_t isRegression=kFALSE, Bool_t greyScale=kFALSE, Bool_t useTMVAStyle=kTRUE)
Definition: file.py:1