Logo ROOT  
Reference Guide
correlations.cxx
Go to the documentation of this file.
1#include "TMVA/correlations.h"
2#include "TMVA/Config.h"
3
4
5#include "TH1.h"
6#include "TH2.h"
7#include "TPaletteAxis.h"
8
9// this macro plots the correlation matrix of the various input
10// variables used in TMVA (e.g. running TMVAnalysis.C). Signal and
11// Background are plotted separately
12
13// input: - Input file (result from TMVA),
14// - use of colors or grey scale
15// - use of TMVA plotting TStyle
16void TMVA::correlations(TString dataset, TString fin , Bool_t isRegression ,
17 Bool_t /* greyScale */ , Bool_t useTMVAStyle )
18{
19
20 // set style and remove existing canvas'
21 TMVAGlob::Initialize( useTMVAStyle );
22
23 // checks if file with name "fin" is already open, and if not opens one
24 TFile* file = TMVAGlob::OpenFile( fin );
25
26 // signal and background or regression problem
27 Int_t ncls = (isRegression ? 1 : 2 );
28 TString hName[2] = { "CorrelationMatrixS", "CorrelationMatrixB" };
29 if (isRegression) hName[0]= "CorrelationMatrix";
30 const Int_t width = 600;
31 for (Int_t ic=0; ic<ncls; ic++) {
32
33 TH2* h2 = dynamic_cast<TH2*> (file->GetDirectory(dataset.Data())->Get( hName[ic] ));
34 if(!h2) {
35 cout << "Did not find histogram " << hName[ic] << " in " << fin << endl;
36 continue;
37 }
38
39 TCanvas* c = new TCanvas( hName[ic],
40 Form("Correlations between MVA input variables (%s)",
41 (isRegression ? "" : (ic==0 ? "signal" : "background"))),
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 += hName[ic];
87 TMVAGlob::imgconv( c, fname );
88 }
89}
#define c(i)
Definition: RSha256.hxx:101
int Int_t
Definition: RtypesCore.h:41
bool Bool_t
Definition: RtypesCore.h:59
float Float_t
Definition: RtypesCore.h:53
include TDocParser_001 C image html pict1_TDocParser_001 png width
Definition: TDocParser.cxx:121
char * Form(const char *fmt,...)
R__EXTERN TStyle * gStyle
Definition: TStyle.h:407
virtual void SetLabelSize(Float_t size=0.04)
Set size of axis labels.
Definition: TAttAxis.cxx:204
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:46
The Canvas class.
Definition: TCanvas.h:31
A ROOT file is a suite of consecutive data records (TKey instances) with a well defined format.
Definition: TFile.h:48
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")
Set option(s) to draw axis with labels.
Definition: TH1.cxx:5214
TAxis * GetXaxis()
Get the behaviour adopted by the object about the statoverflows. See EStatOverflows for more informat...
Definition: TH1.h:316
TAxis * GetYaxis()
Definition: TH1.h:317
TList * GetListOfFunctions() const
Definition: TH1.h:239
virtual void Draw(Option_t *option="")
Draw this histogram with options.
Definition: TH1.cxx:2998
Service class for 2-Dim histogram classes.
Definition: TH2.h:30
virtual TObject * FindObject(const char *name) const
Find an object in this list using its name.
Definition: TList.cxx:575
virtual void AppendPad(Option_t *option="")
Append graphics object to current pad.
Definition: TObject.cxx:105
The palette painting class.
Definition: TPaletteAxis.h:29
virtual void SetLabelSize(Float_t labelsize)
Definition: TPaletteAxis.h:62
Double_t GetX1NDC() const
Definition: TPave.h:59
virtual void SetX1NDC(Double_t x1)
Definition: TPave.h:78
Basic string class.
Definition: TString.h:131
const char * Data() const
Definition: TString.h:364
void SetPaintTextFormat(const char *format="g")
Definition: TStyle.h:364
void SetPalette(Int_t ncolors=kBird, Int_t *colors=0, Float_t alpha=1.)
See TColor::SetPalette.
Definition: TStyle.cxx:1643
Base class for several text objects.
Definition: TText.h:23
virtual void SetNDC(Bool_t isNDC=kTRUE)
Set NDC mode on if isNDC = kTRUE, off otherwise.
Definition: TText.cxx:812
void Initialize(Bool_t useTMVAStyle=kTRUE)
Definition: tmvaglob.cxx:176
TFile * OpenFile(const TString &fin)
Definition: tmvaglob.cxx:192
void imgconv(TCanvas *c, const TString &fname)
Definition: tmvaglob.cxx:212
Config & gConfig()
void correlations(TString dataset, TString fin="TMVA.root", Bool_t isRegression=kFALSE, Bool_t greyScale=kFALSE, Bool_t useTMVAStyle=kTRUE)
Definition: file.py:1