refactor genome.py use (C, 4) to replace (2, N, N) to represent connections

faster, faster and faster!
2023-05-12 00:57:55 +08:00
parent e5fc1167d9
commit 47b1a1dbb2
16 changed files with 363 additions and 419 deletions
--- a/algorithms/neat/genome/activations.py
+++ b/algorithms/neat/genome/activations.py
@@ -23,8 +23,8 @@ def sin_act(z):

@jit
 def gauss_act(z):
-    z = jnp.clip(z, -3.4, 3.4)
-    return jnp.exp(-5 * z ** 2)
+    z = jnp.clip(z * 5, -3.4, 3.4)
+    return jnp.exp(-z ** 2)


@jit