Ignore:
Timestamp:
03/01/06 13:27:34 (19 years ago)
Author:
tbretz
Message:
*** empty log message ***
Location:
trunk/MagicSoft/Mars/mjtrain
Files:
4 edited

Legend:

Unmodified
Added
Removed
  • TabularUnified trunk/MagicSoft/Mars/mjtrain/MJTrainDisp.cc

    r7433 r7552  
    131131    // ------------------------ Train RF --------------------------
    132132    MRanForestCalc rf;
     133    rf.SetNumTrees(fNumTrees);
     134    rf.SetNdSize(fNdSize);
     135    rf.SetNumTry(fNumTry);
    133136    rf.SetNumObsoleteVariables(1);
    134137    rf.SetDisplay(fDisplay);
  • TabularUnified trunk/MagicSoft/Mars/mjtrain/MJTrainEnergy.cc

    r7433 r7552  
    125125    // ------------------------ Train RF --------------------------
    126126    MRanForestCalc rf;
     127    rf.SetNumTrees(fNumTrees);
     128    rf.SetNdSize(fNdSize);
     129    rf.SetNumTry(fNumTry);
    127130    rf.SetNumObsoleteVariables(3);
    128131    rf.SetDisplay(fDisplay);
  • TabularUnified trunk/MagicSoft/Mars/mjtrain/MJTrainRanForest.h

    r7412 r7552  
    1919    TList fTestCuts;
    2020
     21    UShort_t fNumTrees;
     22    UShort_t fNdSize;
     23    UShort_t fNumTry;
     24
    2125    Bool_t WriteDisplay(const char *fname) const;
    2226
     
    2529
    2630public:
    27     MJTrainRanForest() : fDebug(kFALSE) { }
     31    MJTrainRanForest() : fDebug(kFALSE)
     32    {
     33        fNumTrees = 100; //100
     34        fNumTry   = 0;   //3   0 means: in MRanForest estimated best value will be calculated
     35        fNdSize   = 1;   //1
     36    }
    2837
    2938    void SetDebug(Bool_t b=kTRUE) { fDebug = b; }
     
    3847    void AddTestCut(MFilter *f)        { AddCut(fTestCuts, f); }
    3948
     49    void SetNumTrees(UShort_t n=100) { fNumTrees = n; }
     50    void SetNdSize(UShort_t n=5)     { fNdSize   = n; }
     51    void SetNumTry(UShort_t n=0)     { fNumTry   = n; }
     52
    4053    Int_t AddParameter(const char *rule);
    4154
  • TabularUnified trunk/MagicSoft/Mars/mjtrain/MJTrainSeparation.cc

    r7539 r7552  
    468468
    469469    MRanForestCalc rf;
     470    rf.SetNumTrees(fNumTrees);
     471    rf.SetNdSize(fNdSize);
     472    rf.SetNumTry(fNumTry);
    470473    rf.SetNumObsoleteVariables(1);
    471474    rf.SetDebug(fDebug);
Note: See TracChangeset for help on using the changeset viewer.