LCFIPlus  0.6.5
TrainMVA.h
Go to the documentation of this file.
1 // TrainMVA.h
2 
3 #ifndef TrainMVA_h
4 #define TrainMVA_h 1
5 
6 class TFile;
7 class TTree;
8 
9 #include "lcfiplus.h"
10 #include "flavtag.h"
11 
12 #include "TMVA/Types.h"
13 
14 namespace lcfiplus {
15 
21 class TrainMVA : public Algorithm {
22  public:
23  TrainMVA() {}
24  virtual ~TrainMVA() {}
25 
26  void init(Parameters* param);
27  void process();
28  void end();
29 
30 
31  struct InputFileInfo {
32  TString name;
33  TString fileName;
34  TString treeName;
35  TString presel;
36  TFile* file;
37  TTree* tree;
38 
40  InputFileInfo( TString n, TString f, TString t, TString p ) :
41  name(n), fileName(f), treeName(t), presel(p), file(0), tree(0) {}
42  };
43 
44 
45  private:
46  bool _verbose;
47  vector<InputFileInfo> _inputFileInfo;
48  void readInputFileInfo( Parameters* param, TString name );
49 
50  TString _outputDirectory;
51  TString _outputPrefix;
52  TString _treeName;
53 
54  TMVA::Types::EMVA _tmvaBookType;
55  TString _tmvaBookName;
56  TString _tmvaBookOptions;
57  int _skipTrain;
58 
59  vector<FlavtagCategory> _categories;
60 
61  ClassDef(TrainMVA,1);
62 };
63 
64 }
65 
66 #endif
virtual ~TrainMVA()
Definition: TrainMVA.h:24
Definition: TrainMVA.h:31
TFile * file
Definition: TrainMVA.h:36
InputFileInfo(TString n, TString f, TString t, TString p)
Definition: TrainMVA.h:40
Definition: lcfiplus.h:269
Definition: lcfiplus.h:152
Lcfiplus algorithm for training classifications using TMVA.
Definition: TrainMVA.h:21
TString name
Definition: TrainMVA.h:32
void init(Parameters *param)
Definition: TrainMVA.cc:53
TrainMVA()
Definition: TrainMVA.h:23
TTree * tree
Definition: TrainMVA.h:37
InputFileInfo()
Definition: TrainMVA.h:39
void end()
Definition: TrainMVA.cc:151
TString fileName
Definition: TrainMVA.h:33
TString treeName
Definition: TrainMVA.h:34
void process()
Definition: TrainMVA.cc:147
TString presel
Definition: TrainMVA.h:35