| Line | |
|---|
| 1 | #ifndef MARS_MRanForestGrow
|
|---|
| 2 | #define MARS_MRanForestGrow
|
|---|
| 3 |
|
|---|
| 4 | #ifndef MARS_MTask
|
|---|
| 5 | #include "MTask.h"
|
|---|
| 6 | #endif
|
|---|
| 7 |
|
|---|
| 8 | #include "TRandom.h"
|
|---|
| 9 | #include "TMath.h"
|
|---|
| 10 |
|
|---|
| 11 | /////////////////////////////////////////////////////////////////////////////
|
|---|
| 12 | // //
|
|---|
| 13 | // MRanForestGrow //
|
|---|
| 14 | // //
|
|---|
| 15 | // Task to grow (train) a random forest //
|
|---|
| 16 | // //
|
|---|
| 17 | /////////////////////////////////////////////////////////////////////////////
|
|---|
| 18 |
|
|---|
| 19 | #include "MRanForest.h"
|
|---|
| 20 |
|
|---|
| 21 | class MHMatrix;
|
|---|
| 22 | class MParList;
|
|---|
| 23 | class MRanForest;
|
|---|
| 24 |
|
|---|
| 25 | class MRanForestGrow : public MTask
|
|---|
| 26 | {
|
|---|
| 27 | private:
|
|---|
| 28 | MRanTree *fRanTree;
|
|---|
| 29 | MRanForest *fRanForest;
|
|---|
| 30 | MHMatrix *fMGammas; //! Gammas describing matrix
|
|---|
| 31 | MHMatrix *fMHadrons; //! Hadrons (non gammas) describing matrix
|
|---|
| 32 |
|
|---|
| 33 | Int_t fNumTrees;
|
|---|
| 34 | Int_t fNumTry;
|
|---|
| 35 | Int_t fNdSize;
|
|---|
| 36 |
|
|---|
| 37 | public:
|
|---|
| 38 | MRanForestGrow(const char *name=NULL, const char *title=NULL);
|
|---|
| 39 |
|
|---|
| 40 | void SetNumTrees(Int_t n);
|
|---|
| 41 | void SetNumTry(Int_t n);
|
|---|
| 42 | void SetNdSize(Int_t n);
|
|---|
| 43 |
|
|---|
| 44 | Bool_t PreProcess(MParList *pList);
|
|---|
| 45 | Bool_t Process();
|
|---|
| 46 | Bool_t PostProcess();
|
|---|
| 47 |
|
|---|
| 48 | ClassDef(MRanForestGrow, 0) // Task to grow a random forest
|
|---|
| 49 | };
|
|---|
| 50 |
|
|---|
| 51 | #endif
|
|---|
| 52 |
|
|---|
Note:
See
TracBrowser
for help on using the repository browser.
