source: tags/Mars-V2.0/mjtrain/MJTrainDisp.h

Last change on this file was 8719, checked in by tbretz, 17 years ago
*** empty log message ***
File size: 1007 bytes
Line 
1#ifndef MARS_MJTrainDisp
2#define MARS_MJTrainDisp
3
4#ifndef MARS_MJTrainRanForest
5#include "MJTrainRanForest.h"
6#endif
7
8class TCanvas;
9
10class MH3;
11class MDataSet;
12
13class MJTrainDisp : public MJTrainRanForest
14{
15private:
16 static const TString fgTrainParameter;
17
18 TString fTrainParameter;
19 TString fResultFunction;
20
21 Float_t fThetaCut;
22
23 // To be moved to a base class
24 TString fNameOutput;
25
26 void DisplayHist(TCanvas &c, Int_t i, MH3 &mh3) const;
27 void DisplayResult(MH3 &hsize, MH3 &henergy);
28
29public:
30 MJTrainDisp() : fTrainParameter(fgTrainParameter), fResultFunction("x"), fThetaCut(0.215), fNameOutput("Disp") { }
31
32 void SetTrainParameter(const char *txt) { fTrainParameter=txt; }
33
34 void SetThetaCut(Float_t cut=0.215) { fThetaCut=cut; }
35
36 Bool_t Train(const char *out, const MDataSet &set, Int_t num);
37 //Bool_t TrainGhostbuster(const char *out, const MDataSet &set, Int_t num);
38
39 ClassDef(MJTrainDisp, 0)//Class to train Random Forest disp estimator
40};
41
42#endif
Note: See TracBrowser for help on using the repository browser.