Changeset 7178 for trunk/MagicSoft/Mars/mranforest/MRFEnergyEst.h

Timestamp:

07/08/05 18:14:35 (19 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mranforest/MRFEnergyEst.h (modified) (1 diff)

trunk/MagicSoft/Mars/mranforest/MRFEnergyEst.h

@@ -23 +23 @@
     {
         kMean,
-        kMaximum
+        kMaximum,
+        kFit
     };
 private:
-    Int_t        fNumTrees;  // Training parameters
-    Int_t        fNumTry;    // Training parameters
-    Int_t        fNdSize;    // Training parameters
+    Bool_t       fDebug;      // Debugging of eventloop while training on/off
 
-    Bool_t       fDebug;     // Debugging of eventloop while training on/off
-
-    TString      fFileName;
-    TObjArray    fEForests;
+    TString      fFileName;   // File name to forest
+    TObjArray    fEForests;   // List of forests
 
     MDataArray  *fData;       //! Used to store the MDataChains to get the event values
     MParameterD *fEnergyEst;  //! Used to storeestimated energy
 
-    MHMatrix *fTestMatrix;
+    Int_t        fNumTrees;   //! Training parameters
+    Int_t        fNumTry;     //! Training parameters
+    Int_t        fNdSize;     //! Training parameters
 
-    TArrayD   fBinning;
+    MHMatrix    *fTestMatrix; //! Test Matrix
 
     EstimationMode_t fEstimationMode;
 
+    // MRFEnergyEst
+    Int_t ReadForests(MParList &plist);
+
+    // MTask
     Int_t PreProcess(MParList *plist);
     Int_t Process();
 
-    Int_t ReadForests(MParList &plist);
-
+    // MParContainer
     Int_t ReadEnv(const TEnv &env, TString prefix, Bool_t print);
 
 public:
     MRFEnergyEst(const char *name=NULL, const char *title=NULL);
+    ~MRFEnergyEst();
 
+    // Setter for estimation
     void  SetFileName(TString str)     { fFileName = str; }
+    void  SetEstimationMode(EstimationMode_t op) { fEstimationMode = op; }
 
+    // Setter for training
     void  SetNumTrees(Int_t n=-1)      { fNumTrees = n; }
     void  SetNdSize(Int_t n=-1)        { fNdSize   = n; }
     void  SetNumTry(Int_t n=-1)        { fNumTry   = n; }
+    void  SetDebug(Bool_t b=kTRUE)     { fDebug = b; }
 
+    // Train Interface
+    Int_t Train(const MHMatrix &n, const TArrayD &grid);
+
+    // Test Interface
     void  SetTestMatrix(MHMatrix *m=0) { fTestMatrix=m; }
     void  InitMapping(MHMatrix *m=0)   { fTestMatrix=m; }
-
-    void  SetDebug(Bool_t b=kTRUE)     { fDebug = b; }
-
-    void  SetEstimationMode(EstimationMode_t op) { fEstimationMode = op; }
-
-    Int_t Train(const MHMatrix &n, const TArrayD &grid);
 
     ClassDef(MRFEnergyEst, 0) // Task