Changeset 7424 for trunk/MagicSoft/Mars/mranforest/MRanForest.cc

Timestamp:

11/22/05 16:42:11 (19 years ago)

Author:

tbretz

Message:

*** empty log message ***

File:

• trunk/MagicSoft/Mars/mranforest/MRanForest.cc (modified) (3 diffs)

trunk/MagicSoft/Mars/mranforest/MRanForest.cc

@@ -133 +133 @@
 }
 
-MRanTree *MRanForest::GetTree(Int_t i)
-{
-    return (MRanTree*)(fForest->At(i));
+void MRanForest::Print(Option_t *o) const
+{
+    *fLog << inf << GetDescriptor() << ": " << endl;
+    MRanTree *t = GetTree(0);
+    if (t)
+    {
+        *fLog << "Setting up RF for training on target:" << endl;
+        *fLog << " " << t->GetTitle() << endl;
+    }
+    if (fRules)
+    {
+        *fLog << "Following rules are used as input to RF:" << endl;
+        for (Int_t i=0;i<fRules->GetNumEntries();i++)
+            *fLog << " " << i << ") " << (*fRules)[i].GetRule() << endl;
+    }
+    *fLog << "Random forest parameters:" << endl;
+    if (t)
+    {
+        *fLog << " - " << (t->IsClassify()?"classification":"regression") << " tree" << endl;
+        *fLog << " - Number of trys: " << t->GetNumTry() << endl;
+        *fLog << " - Node size: " << t->GetNdSize() << endl;
+    }
+    *fLog << " - Number of trees: " << fNumTrees << endl;
+    *fLog << " - User value: " << fUserVal << endl;
+    *fLog << endl;
 }
 
@@ -335 +357 @@
     for (Int_t i=0;i<dim;i++)
         *fLog << " " << i << ") " << (*fRules)[i].GetRule() << endl;
-
     *fLog << endl;
 
@@ -351 +372 @@
         return kFALSE;
     }
+    fRanTree->SetName(target_rule);
 
     const Int_t tryest = TMath::Nint(TMath::Sqrt(dim));