fixes ml seeding error when more features than individuals

lacava · Oct 17, 2016 · 263be4e · 263be4e
1 parent 09a790b
commit 263be4e
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/few/_version.py b/few/_version.py
@@ -6,4 +6,4 @@
 
 """
 
-__version__ = '0.0.6'
+__version__ = '0.0.7'
diff --git a/few/few.py b/few/few.py
@@ -496,11 +496,12 @@ def init_pop(self):
                         raise(AttributeError)
                 except Exception: # seed pop with raw features
                      for i,p in it.zip_longest(range(self._training_features.shape[1]),pop.individuals,fillvalue=None):
-                         if i is not None:
-                             p.stack = [('x',0,i)]
-                         else:
-                             make_program(p.stack,self.func_set,self.term_set,np.random.randint(self.min_depth,self.max_depth+1),self.otype)
-                             p.stack = list(reversed(p.stack))
+                        if p is not None:
+                            if i is not None:
+                                p.stack = [('x',0,i)]
+                            else:
+                                make_program(p.stack,self.func_set,self.term_set,np.random.randint(self.min_depth,self.max_depth+1),self.otype)
+                                p.stack = list(reversed(p.stack))
 
             # print initial population
             if self.verbosity > 2: print("seeded initial population:",stacks_2_eqns(pop.individuals))