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