Changeset 7130 for trunk/MagicSoft/Mars/mranforest/MRanForestGrow.h

Timestamp:

06/03/05 18:02:36 (19 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mranforest/MRanForestGrow.h (modified) (3 diffs)

trunk/MagicSoft/Mars/mranforest/MRanForestGrow.h

@@ -2 +2 @@
 #define MARS_MRanForestGrow
 
-/////////////////////////////////////////////////////////////////////////////
-//                                                                         //
-// MRanForestGrow                                                          //
-//                                                                         //
-// Task to grow (train) a random forest                                    //
-//                                                                         //
-/////////////////////////////////////////////////////////////////////////////
-
-#ifndef MARS_MTask
-#include "MTask.h"
+#ifndef MARS_MRead
+#include "MRead.h"
 #endif
 
@@ -19 +11 @@
 class MRanTree;
 
-class MRanForestGrow : public MTask
+class MRanForestGrow : public MRead
 {
 private:
@@ -35 +27 @@
     Int_t PostProcess();
 
+    UInt_t  GetEntries()            { return fNumTrees; }
+    TString GetFileName() const     { return "MRanForestGrow"; }
+    TString GetFullFileName() const { return "MRanForestGrow"; }
+
 public:
     MRanForestGrow(const char *name=NULL, const char *title=NULL);
 
-    void SetNumTrees(Int_t n){    fNumTrees=n;}
-    void SetNumTry(Int_t n)  {    fNumTry=n;  }
-    void SetNdSize(Int_t n)  {    fNdSize=n;  }
+    void SetNumTrees(Int_t n=-1) { fNumTrees=n>0?n:100; }
+    void SetNumTry(Int_t   n=-1) { fNumTry  =n>0?n:  3; }
+    void SetNdSize(Int_t   n=-1) { fNdSize  =n>0?n:  1; }
 
     ClassDef(MRanForestGrow, 0) // Task to grow a random forest