hyper neat

examples/xor.py

@@ -1,10 +1,9 @@
 import jax
 import numpy as np
 
-from config import Config, BasicConfig
+from config import Config, BasicConfig, NeatConfig
 from pipeline import Pipeline
-from algorithm.neat.gene import NormalGene, NormalGeneConfig
-from algorithm.neat.neat import NEAT
+from algorithm import NEAT, NormalGene, NormalGeneConfig
 
 xor_inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
 xor_outputs = np.array([[0], [1], [1], [0]], dtype=np.float32)
@@ -23,8 +22,14 @@ def evaluate(forward_func):
 
 if __name__ == '__main__':
     config = Config(
-        basic=BasicConfig(fitness_target=4),
-        gene=NormalGeneConfig()
+        basic=BasicConfig(
+            fitness_target=3.99999,
+            pop_size=10000
+        ),
+        neat=NeatConfig(
+            maximum_nodes=50,
+            maximum_conns=100,
+        )
     )
     algorithm = NEAT(config, NormalGene)
     pipeline = Pipeline(config, algorithm)