tony/tensorneat-mend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

complete normal neat algorithm

Browse Source
This commit is contained in:
wls2002
2023-07-18 23:55:36 +08:00
parent 40cf0b6fbe
commit 0a2a9fd1be
26 changed files with 880 additions and 251 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
examples/xor_test.py
View File

@@ -1,17 +1,32 @@
import jax
import numpy as np
from algorithm.config import Configer
from algorithm.neat import NEAT
from algorithm.neat import NEAT, NormalGene, Pipeline
from algorithm.neat.genome import create_mutate
xor_inputs = np.array([[0, 0], [0, 1], [1, 0], [1, 1]], dtype=np.float32)
def single_genome(func, nodes, conns):
t = NormalGene.forward_transform(nodes, conns)
out1 = func(xor_inputs[0], t)
out2 = func(xor_inputs[1], t)
out3 = func(xor_inputs[2], t)
out4 = func(xor_inputs[3], t)
print(out1, out2, out3, out4)
if __name__ == '__main__':
config = Configer.load_config()
neat = NEAT(config)
neat = NEAT(config, NormalGene)
randkey = jax.random.PRNGKey(42)
state = neat.setup(randkey)
state = neat.mutate(state)
print(state)
pop_nodes, pop_conns = state.pop_nodes, state.pop_conns
print(neat.distance(state, pop_nodes[0], pop_conns[0], pop_nodes[1], pop_conns[1]))
print(neat.crossover(state, pop_nodes[0], pop_conns[0], pop_nodes[1], pop_conns[1]))
forward_func = NormalGene.create_forward(config)
mutate_func = create_mutate(config, NormalGene)
nodes, conns = state.pop_nodes[0], state.pop_conns[0]
single_genome(forward_func, nodes, conns)
nodes, conns = mutate_func(state, randkey, nodes, conns, 10000)
single_genome(forward_func, nodes, conns)