Changeset 2730 for trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization

Timestamp:

02/01/10 16:43:56 (15 years ago)

Author:

gkronber

Message:

Made PTC2 operator more robust. #859 (PTC2 initialization operator fails if it is used with restrictive function libraries)

File:

• trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization/ProbabilisticTreeCreator.cs (modified) (3 diffs)

trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization/ProbabilisticTreeCreator.cs

@@ -24 +24 @@
 using HeuristicLab.Random;
 using HeuristicLab.GP.Interfaces;
+using System;
 
 namespace HeuristicLab.GP.Operators {
@@ -58 +59 @@
     public static IFunctionTree Create(IRandom random, FunctionLibrary funLib, int minSize, int maxSize, int maxHeight) {
       int treeSize = random.Next(minSize, maxSize);
-      IFunctionTree root;
+      IFunctionTree root = null;
       int tries = 0;
       TreeGardener gardener = new TreeGardener(random, funLib);
       do {
-        root = gardener.PTC2(treeSize, maxHeight);
+        try {
+          root = gardener.PTC2(treeSize, maxHeight);
+        }
+        catch (ArgumentException) {
+          // try a different size
+          treeSize = random.Next(minSize, maxSize);
+          tries = 0;
+        }
         if (tries++ >= MAX_TRIES) {
           // try a different size
@@ -68 +76 @@
           tries = 0;
         }
-      } while (root.GetSize() > maxSize || root.GetHeight() > maxHeight);
+      } while (root == null || root.GetSize() > maxSize || root.GetHeight() > maxHeight);
       return root;
     }