This commit is contained in:
root
2024-07-10 16:50:36 +08:00
parent 4cdac932d3
commit 51cb4695af
8 changed files with 71 additions and 72 deletions

View File

@@ -1,7 +1,7 @@
import jax, jax.numpy as jnp
from tensorneat.algorithm import NEAT
from tensorneat.algorithm.neat import DefaultGenome, RecurrentGenome
from tensorneat.genome import DefaultGenome, RecurrentGenome
key = jax.random.key(0)
genome = DefaultGenome(num_inputs=5, num_outputs=3, max_nodes=100, max_conns=500, init_hidden_layers=(1, 2 ,3))