complete fully stateful!

use black to format all files!

tensorneat/examples/func_fit/xor.py

@@ -4,7 +4,7 @@ from algorithm.neat import *
 from problem.func_fit import XOR3d
 from utils import Act
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     pipeline = Pipeline(
         algorithm=NEAT(
             species=DefaultSpecies(
@@ -18,22 +18,22 @@ if __name__ == '__main__':
                         activation_options=(Act.tanh,),
                     ),
                     output_transform=Act.sigmoid,  # the activation function for output node
+                    mutation=DefaultMutation(
+                        node_add=0.05,
+                        conn_add=0.2,
+                        node_delete=0,
+                        conn_delete=0,
+                    ),
                 ),
                 pop_size=10000,
                 species_size=10,
                 compatibility_threshold=3.5,
                 survival_threshold=0.01,  # magic
             ),
-            mutation=DefaultMutation(
-                node_add=0.05,
-                conn_add=0.2,
-                node_delete=0,
-                conn_delete=0,
-            )
         ),
         problem=XOR3d(),
         generation_limit=10000,
-        fitness_target=-1e-8
+        fitness_target=-1e-8,
     )
 
     # initialize state