Implemented writing of DefaultStagnation and DefaultReproduction configuration to file.

Author: CodeReclaimers

neat/reproduction.py

@@ -20,11 +20,18 @@ class DefaultReproduction(object):
     @classmethod
     def parse_config(cls, param_dict):
         config = {'elitism': 1,
-                       'survival_threshold': 0.2}
+                  'survival_threshold': 0.2}
         config.update(param_dict)
 
         return config
 
+    @classmethod
+    def write_config(cls, f, param_dict):
+        elitism = param_dict.get('elitism', 1)
+        f.write('elitism            = {}\n'.format(elitism))
+        survival_threshold = param_dict.get('survival_threshold', 0.2)
+        f.write('survival_threshold = {}\n'.format(survival_threshold))
+
     def __init__(self, config, reporters, stagnation):
         self.elitism = int(config.get('elitism'))
         self.survival_threshold = float(config.get('survival_threshold'))

neat/stagnation.py

@@ -15,6 +15,13 @@ def parse_config(cls, param_dict):
 
         return config
 
+    @classmethod
+    def write_config(cls, f, config):
+        fitness_func = config.get('species_fitness_func', 'mean')
+        f.write('species_fitness_func = {}\n'.format(fitness_func))
+        max_stagnation = config.get('max_stagnation', 15)
+        f.write('max_stagnation       = {}\n'.format(max_stagnation))
+
     def __init__(self, config, reporters):
         self.max_stagnation = int(config.get('max_stagnation'))
         self.species_fitness = config.get('species_fitness_func')