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

make fully stateful in module ga.

Browse Source
This commit is contained in:
wls2002
2024-05-25 16:47:25 +08:00
parent 485d481745
commit c0a7503168
5 changed files with 24 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tensorneat/algorithm/neat/ga/crossover/default.py
View File

@@ -5,12 +5,12 @@ from .base import BaseCrossover
class DefaultCrossover(BaseCrossover):
def __call__(self, randkey, genome, nodes1, conns1, nodes2, conns2):
def __call__(self, state, key, genome, nodes1, conns1, nodes2, conns2):
"""
use genome1 and genome2 to generate a new genome
notice that genome1 should have higher fitness than genome2 (genome1 is winner!)
"""
randkey_1, randkey_2, key = jax.random.split(randkey, 3)
randkey_1, randkey_2, key = jax.random.split(key, 3)
# crossover nodes
keys1, keys2 = nodes1[:, 0], nodes2[:, 0]