root/TMVA/network.h

0001 #ifndef network__HH
0002 #define network__HH
0003 #include "TArrow.h"
0004 #include "TEllipse.h"
0005 #include "TPaveLabel.h"
0006 #include "TCanvas.h"
0007 #include "TH2F.h"
0008 #include "TFile.h"
0009 #include "TString.h"
0010 #include "TDirectory.h"
0011 #include "TKey.h"
0012 #include "TText.h"
0013
0014 #include "tmvaglob.h"
0015 namespace TMVA{
0016
0017    // this macro prints out a neural network generated by MethodMLP graphically
0018    // @author: Matt Jachowski, jachowski@stanford.edu
0019
0020    void     draw_layer_labels( Int_t nLayers );
0021    void     draw_input_labels(TString dataset,Int_t nInputs, Double_t* cy,
0022                               Double_t rad, Double_t layerWidth);
0023    void     draw_layer       (TString dataset, TCanvas* c, TH2F* h, Int_t iHist, Int_t nLayers, Double_t maxWeight );
0024    void     draw_synapse     ( Double_t cx1, Double_t cy1, Double_t cx2, Double_t cy2,
0025                                Double_t  rad1, Double_t rad2, Double_t weightNormed );
0026    TString* get_var_names    (TString dataset, Int_t nVars );
0027
0028    void draw_network(TString dataset, TFile* f, TDirectory* d, const TString& hName = "weights_hist",
0029                      Bool_t movieMode = kFALSE, const TString& epoch = "" );
0030
0031    void draw_activation(TCanvas* c, Double_t cx, Double_t cy,
0032                         Double_t radx, Double_t rady, Int_t whichActivation);
0033    void network(TString dataset, TString fin = "TMVA.root", Bool_t useTMVAStyle = kTRUE );
0034
0035 }
0036 #endif